https://www.acmicpc.net/problem/11779
import sys
import heapq
INF = int(1e9)
input = sys.stdin.readline
n = int(input())
m = int(input())
graph = [[] for i in range(n + 1)]
for i in range(m):
u, v, w = map(int, input().split())
graph[u].append([v, w])
start, stop = map(int, input().split())
def dijkstra(start):
heap = []
distance = [INF] * (n + 1)
distance_path = [[] for i in range(n + 1)]
heapq.heappush(heap, [0, start, [start]])
distance[start] = 0
while heap:
dist, now, path = heapq.heappop(heap)
if distance[now] < dist:
continue
for i in graph[now]:
cost = dist + i[1]
if cost < distance[i[0]]:
distance[i[0]] = cost
distance_path[i[0]] = path + [i[0]]
heapq.heappush(heap, [cost, i[0], distance_path[i[0]]])
return distance, distance_path
distance, distance_path = dijkstra(start)
print(distance[stop])
print(len(distance_path[stop]))
for i in distance_path[stop]:
print(i, end=" ")
다익스트라를 활용해서 풀면되는데 다른 문제들과 차이점은 경로를 출력해야한다는 점이다
다익스트라 메소드 안에 distance_path를 정의하고, heap에 넣는 값에 경로 값도 추가한다
이러면 최소 비용의 해가 구해질 때 path도 같이 업데이트 되기 때문에 결과적으로 테이블은 최소 비용, 최소 비용의 경로 이렇게 두가지가 구해진다