Update 4.py

ndb796 · web-flow · commit a4bb5e93e9d3 · 2020-06-18T08:10:27.000+09:00
diff --git a/9/4.py b/9/4.py
@@ -1,37 +1,38 @@
+INF = int(1e9) # 무한을 의미하는 값으로 10억을 설정합니다.
+
 # 노드의 개수 및 간선의 개수를 입력 받습니다.
 n, m = map(int, input().split())
-# 2차원 리스트를 만들고, 모든 값을 무한으로 초기화합니다.
-adj = [[1e9] * (n + 1) for _ in range(n + 1)]
+# 2차원 리스트(그래프 표현)를 만들고, 모든 값을 무한으로 초기화합니다.
+graph = [[INF] * (n + 1) for _ in range(n + 1)]
 
 # 자기 자신에서 자기 자신으로 가는 비용은 0으로 초기화합니다.
 for a in range(1, n + 1):
     for b in range(1, n + 1):
         if a == b:
-            adj[a][b] = 0
+            graph[a][b] = 0
 
 # 각 간선에 대한 정보를 입력 받아, 그 값으로 초기화합니다.
 for _ in range(m):
     # A와 B가 서로에게 가는 비용은 1이라고 설정합니다.
     a, b = map(int, input().split())
-    adj[a][b] = 1
-    adj[b][a] = 1
+    graph[a][b] = 1
+    graph[b][a] = 1
 
 # 거쳐 갈 노드 x와 최종 목적지 노드 k를 입력 받습니다.
 x, k = map(int, input().split())
 
-# 플로이드 워셜 알고리즘을 정의합니다.
-def solve():
-    for k in range(1, n + 1):
-        for a in range(1, n + 1):
-            for b in range(1, n + 1):
-                adj[a][b] = min(adj[a][b], adj[a][k] + adj[k][b])
-
-# 플로이드 워셜 알고리즘을 수행합니다.
-solve()
+# 점화식에 따라 플로이드 워셜 알고리즘을 수행합니다.
+for k in range(1, n + 1):
+    for a in range(1, n + 1):
+        for b in range(1, n + 1):
+            graph[a][b] = min(graph[a][b], graph[a][k] + graph[k][b])
 
 # 수행된 결과를 출력합니다.
-distance = adj[1][k] + adj[k][x]
+distance = graph[1][k] + graph[k][x]
+
+# 도달할 수 없는 경우, -1을 출력합니다.
 if distance >= 1e9:
     print("-1")
+# 도달할 수 있다면, 최단 거리를 출력합니다.
 else:
     print(distance)
