[백준] (1753) 최단경로 [Python]

문제 링크

https://www.acmicpc.net/problem/1753

문제

방향그래프가 주어지면 주어진 시작점에서 다른 모든 정점으로의 최단 경로를 구하는 프로그램을 작성하시오. 단, 모든 간선의 가중치는 10 이하의 자연수이다.

• 입력
  첫째 줄에 정점의 개수 V와 간선의 개수 E가 주어진다. (1 ≤ V ≤ 20,000, 1 ≤ E ≤ 300,000) 모든 정점에는 1부터 V까지 번호가 매겨져 있다고 가정한다. 둘째 줄에는 시작 정점의 번호 K(1 ≤ K ≤ V)가 주어진다. 셋째 줄부터 E개의 줄에 걸쳐 각 간선을 나타내는 세 개의 정수 (u, v, w)가 순서대로 주어진다. 이는 u에서 v로 가는 가중치 w인 간선이 존재한다는 뜻이다. u와 v는 서로 다르며 w는 10 이하의 자연수이다. 서로 다른 두 정점 사이에 여러 개의 간선이 존재할 수도 있음에 유의한다.
• 출력
  첫째 줄부터 V개의 줄에 걸쳐, i번째 줄에 i번 정점으로의 최단 경로의 경로값을 출력한다. 시작점 자신은 0으로 출력하고, 경로가 존재하지 않는 경우에는 INF를 출력하면 된다.

예제 입력	예제 출력
5 6 1 5 1 1 1 2 2 1 3 3 2 3 4 2 4 5 3 4 6	0 2 3 7 INF

---

코드

import sys
input = sys.stdin.readline
import heapq

v, e = map(int, input().split())
k = int(input())
arr = [[] for _ in range(v+1)]
for _ in range(e):
    a, b, c = map(int, input().split())
    arr[a].append((b, c))

def dijkstra(arr, start):
    D = [float("inf")] * (v+1)
    D[start] = 0
    queue = [(0, start)]

    while queue:
        now_dist, now_node = heapq.heappop(queue)

        if now_dist > D[now_node]:
            continue

        for node, weight in arr[now_node]:
            cost = now_dist + weight
            if cost < D[node]:
                D[node] = cost
                heapq.heappush(queue, (cost, node))

    return D[1:]

result = dijkstra(arr, k)
for i in result:
    if i == float("inf"):
        print("INF")
    else:
        print(i)

 

기본적인 다익스트라 문제로, 다익스트라 알고리즘의 개념과 원리를 이해하고 있다면 쉽게 풀 수 있는 문제다.