정점의 개수 v, 간선의 개수 e, 시작 정점 k를 입력받은 후 각 ‘출발지 도착지 거리’를 u, v, w로 입력받은 후 1부터 v까지 모든 정점을 가는 데 걸리는 최\b단 경로 값을 출력하는 다익스트라 문제. graph[출발지] = {도착지: 거리, 도착지1: 거리}와 같이 입력받아 사용하였다. 두 풀이 모두 힙을 사용하여 최소 거리를 먼저 연산하였다. 첫번째 풀이는 큐에 [거리, 도착지, 출발지]와 같이 넣어 출발지가 시작 노드가 아닐 때 dist[출발지] 값과 거리를 계산하여 넣어 풀었으나 시간초과 결과를 받게 되었다. 두번째 풀이는 큐에 넣는 연산을 하는 반복문에서 dist에 값을 할당하여 풀었다. 자세한 풀이는 주석에 설명해 놓았다.
\n# import heapq\n# import sys\n# input = sys.stdin.readline\n# INF = sys.maxsize\n#\n# def dijkstra(k, graph):\n# dist = [INF for _ in range(len(graph))]\n# q = []\n# heapq.heappush(q, [0, k, k])\n# weight = 0\n# while q:\n# weight, here, start = heapq.heappop(q)\n# if start == k:\n# dist[here] = min(dist[here], weight)\n# else:\n# dist[here] = min(dist[here], weight + dist[start])\n#\n# for there, distance in graph[here].items():\n# heapq.heappush(q, [distance, there, here])\n#\n# return dist\n#\n#\n# v, e = map(int, input().split())\n# graph = [{} for _ in range(v+1)]\n# k = int(input())\n# for _ in range(e):\n# u, v, w = map(int, input().split())\n# if v in graph[u]:\n# graph[u][v] = min(w, graph[u][v])\n# else:\n# graph[u][v] = w\n#\n# dist = dijkstra(k, graph)\n# for i in dist[1:]:\n# print(i if i != INF else 'INF')\n\n\nimport heapq # 거리가 짧은 순으로 계산해야하기 때문에 힙 사용\nimport sys\ninput = sys.stdin.readline\nINF = sys.maxsize\n\ndef dijkstra(k, graph):\n # 노드의 수 + 1 만큼 최대 크기로 이루어진 거리를 저장할 리스트\n dist = [INF for _ in range(len(graph))]\n # 시작 노드의 거리는 0으로\n dist[k] = 0\n\n q = []\n # 거리, 행선지 순으로 리스트에 힙 연산으로 push\n heapq.heappush(q, [0, k])\n while q:\n # 거리 weight, 행선지 here을 힙 연산으로 pop\n weight, here = heapq.heappop(q)\n\n # graph[행선지] = {도착지: 거리} 순으로 저장돼 있음\n for there, distance in graph[here].items():\n # 다음 간선까지 길이는 현재 길이 + 저장된 거리\n next_dist = distance + weight\n\n # 현재 dist에 저장된 거리보다 작을 시\n if dist[there] > next_dist:\n # 저장 후 [거리, 행선지]를 리스트에 힙 연산으로 push\n dist[there] = next_dist\n heapq.heappush(q, [next_dist, there])\n\n return dist\n\n\nv, e = map(int, input().split())\n# graph[출발지] = {도착지: 거리, 도착지1: 거리1}\ngraph = [{} for _ in range(v+1)]\nk = int(input())\nfor _ in range(e):\n u, v, w = map(int, input().split())\n if v in graph[u]:\n graph[u][v] = min(w, graph[u][v])\n else:\n graph[u][v] = w\n\ndist = dijkstra(k, graph)\nfor i in dist[1:]:\n print(i if i != INF else 'INF')