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

 

1916번: 최소비용 구하기

첫째 줄에 도시의 개수 N(1 ≤ N ≤ 1,000)이 주어지고 둘째 줄에는 버스의 개수 M(1 ≤ M ≤ 100,000)이 주어진다. 그리고 셋째 줄부터 M+2줄까지 다음과 같은 버스의 정보가 주어진다. 먼저 처음에는 그

www.acmicpc.net

풀이

1, 양방향 문제이다. INF는 10억이기에 int(1e9)로 해주었다.

2, 다익스트라 적용

# 1916번 최소비용 구하기
import heapq
import sys
input = sys.stdin.readline
INF = int(1e9)

def dijkstra(start, target):
    q = []
    # heapq.heappush(dist, node)
    heapq.heappush(q, (0, start))
    distance[start] = 0
    while q:
        dist, now = heapq.heappop(q)
        if distance[now] < dist:
            continue
        # vertex, weight
        for v, w in graph[now]:
            cost = dist + w
            if cost < distance[v]:
                distance[v] = cost
                heapq.heappush(q, (cost, v))
    print(distance[target])


city = int(input())
bus = int(input())
distance = [INF] * (city+1)
graph = [[] for i in range(city+1)]
for _ in range(bus):
    a, b, c = map(int, input().split())
    graph[a].append((b,c))
    graph[b].append((a,c))
s, t = map(int, input().split())
dijkstra(s, t)

출력결과

개발자 성현