Baekjoon1916-최소비용 구하기
백준 사이트 1916 - 최소비용 구하기 문제입니다.
1. 문제
https://www.acmicpc.net/problem/1916
2. Input , Output
3. 분류 및 난이도
다익스트라 문제입니다.
4. 생각한 것들
- 다익스트라 연습하기 위해 푼 문제
5. code
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
import heapq
import sys
input = sys.stdin.readline
N = int(input())
M = int(input())
graph = [[] for i in range(N+1)
res = [int(1e9)] * (N + 1)
for i in range(M) :
tmp = input().split(' ')
graph[int(tmp[0])].append((int(tmp[1]),int(tmp[2])))
start,end = map(int,input().split(' '))
def dijkstra(first) :
q = []
#first-=1
heapq.heappush(q,(0,first))
res[first] = 0
while q :
dis,current = heapq.heappop(q)
if res[current] < dis : continue
for i in graph[current] :
nextdis = dis + i[1]
if nextdis < res[i[0]] :
res[i[0]] = nextdis
heapq.heappush(q,(nextdis,i[0]))
dijkstra(start)
print(res[end])
6. 후기
This post is licensed under CC BY 4.0 by the author.