n개의 도시, m개의 버스가 있다. 버스는 출발, 도착, 비용을 갖는다. 시작 도시와 도착 도시가 주어질 때, 최소 비용으로 이동했을 시 총 비용, 경로에 포함돼 있는 도시의 수, 경로를 방문하는 도시를 순서대로 출력하는 문제. 힙 자료구조를 이용한 다익스트라 방법을 이용하여 최소 비용을 구함과 동시에 방문지를 저장할 배열을 이용하여 초기화, 지금까지 방문했던 곳 추가, 다음 도시 추가 연산을 통해 풀었다.
\nimport heapq\nimport sys\nINF = sys.maxsize\ninput = sys.stdin.readline\n\ndef solve(n, graph, S, E):\n dist = [INF for _ in range(n+1)]\n visited = [[] for _ in range(n+1)]\n q = []\n\n # cost, current position, visited position\n heapq.heappush(q, [0, S])\n dist[S] = 0\n visited[S] = [S]\n\n while q:\n cost, now = heapq.heappop(q)\n\n if cost > dist[now]:\n continue\n\n for np, tc in graph[now]:\n nc = cost + tc\n if nc < dist[np]:\n heapq.heappush(q, [nc, np])\n dist[np] = nc\n\n visited[np] = []\n for v in visited[now]:\n visited[np].append(v)\n visited[np].append(np)\n\n return dist[E], visited[E]\n\nn = int(input())\nm = int(input())\ngraph = [[] for _ in range(n+1)]\n\nfor _ in range(m):\n s, e, c = map(int, input().split())\n graph[s].append([e, c])\n\nS, E = map(int, input().split())\n\nd, v = solve(n, graph, S, E)\nprint(d)\nprint(len(v))\nprint(*v)