728x90
반응형
1916번: 최소비용 구하기
첫째 줄에 도시의 개수 N(1 ≤ N ≤ 1,000)이 주어지고 둘째 줄에는 버스의 개수 M(1 ≤ M ≤ 100,000)이 주어진다. 그리고 셋째 줄부터 M+2줄까지 다음과 같은 버스의 정보가 주어진다. 먼저 처음에는 그
www.acmicpc.net
import heapq
import sys
input = sys.stdin.readline
n = int(input()) # 도시
m = int(input()) # 버스
graph = [ [] for _ in range(n+1)]
for i in range(m):
s,e,c = map(int,input().split())
graph[s].append((e,c)) # 도착, 비용
s, e = map(int,input().split()) # 출발 도착
def dij(start):
q = []
dist = [1e9]*(n+1)
heapq.heappush(q, (0,start) )
dist[start]=0
while q :
d, now = heapq.heappop(q)
if dist[now]< d : continue
for i in graph[now]:
cost = d + i[1]
if cost < dist[i[0]]:
dist[i[0]] = cost
heapq.heappush(q, (cost,i[0]))
return dist
result = dij(s)
print(result[e])
기본 다익스트라 구현법 사용!
728x90
반응형
'백준 > 최단경로' 카테고리의 다른 글
# 1238 파티 (0) | 2021.04.10 |
---|---|
# 18352 특정 거리의 도시 찾기 (0) | 2021.04.10 |
# 1504 특정한 최단 경로 (0) | 2021.04.10 |
# 1753 최단 경로 파이썬 (0) | 2021.04.09 |