[Baekjoon/python] 최단경로 #1753

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

1753번: 최단경로

문제

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

코드

• 다익스트라 알고리즘 적용

import sys
import heapq
from collections import defaultdict

input = sys.stdin.readline

V, E = map(int, input().split())
src = int(input())
graph = defaultdict(list)
for _ in range(E):
    u, v, w = map(int, input().split())
    graph[u].append((w, v))

def dijkstra(src):
    heap = [(0, src)]
    min_dist = [float('inf')] * (V + 1)
    visited = set()
    min_dist[src] = 0

    while heap:
        cur_w, cur_v = heapq.heappop(heap)
        if cur_v in visited:
            continue
        visited.add(cur_v)
        min_dist[cur_v] = min(min_dist[cur_v], cur_w)
        for nxt_w, nxt_v in graph[cur_v]:
            if nxt_v not in visited:
                w = nxt_w + cur_w
                heapq.heappush(heap, (w, nxt_v))
    return min_dist[1:]

for i in dijkstra(src):
    if i == float('inf'):
        print('INF')
    else:
        print(i)