b1238 - 파티 : 최소힙 다익스트라

Author: BeomSeokYu

Baekjoon/Python/ShortestPath/1238번： 파티/파티.py

@@ -17,35 +17,45 @@
 각 노드가 노드 X를 거쳐 돌아오는 최단 거리 중 그 거리가 가장 긴 거리 출력
 
 플루이드 워셜 이용 모든 정점에 대한 최단거리 정의 -> 시간 초과 -> 1000 ** 3
+-> 다익스트라를 이용해야 함
 '''
 
-import sys
+import sys, heapq
 input = sys.stdin.readline
 
-INF = int(1e9) # 무한을 의미하는 상수
-
-def floyd_warshall(graph, n):
-	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])
+INF = int(1e9) #무한을 의미하는 상수
+
+def dijkstra(graph, start, distance):
+	q = []
+	heapq.heappush(q, (0, start))
+	distance[start] = 0
+	while q:
+		dist, now = heapq.heappop(q)
+		if distance[now] < dist:
+			continue
+		for i in graph[now]:
+			cost = distance[now] + i[1]
+			if cost < distance[i[0]]:
+				distance[i[0]] = cost
+				heapq.heappush(q, (cost, i[0]))
 
 N, M, X = map(int, input().split())
 
-g = [[INF] * (N + 1) for _ in range(N + 1)]
-
-# 자기 자신에서 자기 자신으로 가는 비용은 0으로 초기화
-for a in range(1, N + 1):
-	g[a][a] = 0
+g = [[] for _ in range(N + 1)]
+distance = [[INF] * (N + 1) for _ in range(N + 1)]
 
 for i in range(M):
   s, e, t = map(int, input().split())
-  g[s][e] = t
+  g[s].append((e, t))
 
-floyd_warshall(g, N)
+dijkstra(g, X, distance[X])
+for i in range(1, N + 1):
+  dijkstra(g, i, distance[i])
 
 cost = [0 for _ in range(N + 1)]
 for i in range(1, N + 1):
-  cost[i] = g[i][X] + g[X][i]
+  cost[i] = distance[X][i] + distance[i][X]
+  if (cost[i] > INF) :
+    cost[i] = -1
 
 print(max(cost))