[백준] 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를 출력하면 된다.

풀이

# [백준] 1753번 최단 경로
import heapq
import sys

input = sys.stdin.readline
v,e = map(int,input().split())

start = int(input())

INF =int(1e9)

graph = [[] for _ in range(v+1)]

distance = [INF] * (v+1)

for _ in range(e):
    u,v,w = map(int,input().split())
    graph[u].append((v,w))


def dijkstra(start):
    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 = dist + i[1]
            if cost < distance[i[0]]:
                distance[i[0]] = cost
                heapq.heappush(q,(cost, i[0]))
dijkstra(start)

for dist in distance[1:]:
    if dist == INF:
        print("INF")
    else:
        print(dist)

해당 문제는 다익스트라 알고리즘을 적용하면 쉽게 풀린다. 쉽게 말해서 하나의 정점에 대해 각 노드에 대한 최단 거리를 얻기 위한 알고리즘이다. 여기에 대해 코드가 좀 길어서 외우는 게 쉽지 않아보일수도 있지만, 정작 이해를 하고 코드를 작성하면 금방 작성할 수 있다. 물론 여러 번 연습을 해야 겠지만.

후기

• 다익스트라 알고리즘도 습득해서 기분이 좋다!