[BAEKJOON]백준 1916번 최소비용 구하기 문제 파이썬(Python) 풀이 - 그래프(Graph), 다익스트라(Dijkstra) 알고리즘(Algorithm), 최소 힙(heap) 자료구조

문제

N개의 도시가 있다. 그리고 한 도시에서 출발하여 다른 도시에 도착하는 M개의 버스가 있다. 우리는 A번째 도시에서 B번째 도시까지 가는데 드는 버스 비용을 최소화 시키려고 한다. A번째 도시에서 B번째 도시까지 가는데 드는 최소비용을 출력하여라. 도시의 번호는 1부터 N까지이다.

입력

N개의 도시가 있다. 그리고 한 도시에서 출발하여 다른 도시에 도착하는 M개의 버스가 있다. 우리는 A번째 도시에서 B번째 도시까지 가는데 드는 버스 비용을 최소화 시키려고 한다. A번째 도시에서 B번째 도시까지 가는데 드는 최소비용을 출력하여라. 도시의 번호는 1부터 N까지이다.

출력

첫째 줄에 출발 도시에서 도착 도시까지 가는데 드는 최소 비용을 출력한다.

예제

입력1

5
8
1 2 2
1 3 3
1 4 1
1 5 10
2 4 2
3 4 1
3 5 1
4 5 3
1 5

출력1

4

코드

from sys import stdin
import heapq
from collections import defaultdict


class Solution:
    def minimalCost(self, start: int, end: int, graph):
        heap = []
        heapq.heappush(heap, (0, start))
        dist = [False] * 1005
        while heap:
            w, u = heapq.heappop(heap)
            if dist[u] is False:
                dist[u] = w
                for v, tmp in graph[u]:
                    heapq.heappush(heap, (w + tmp, v))
        print(dist[end])


graph = defaultdict(list)
N = int(stdin.readline())
M = int(stdin.readline())
for i in range(M):
    u, v, w = map(int, stdin.readline().split())
    graph[u].append((v, w))
start, end = map(int, stdin.readline().split())
Solution().minimalCost(start, end, graph)