[백준 12834] 주간 미팅

Junyoung Park·2022년 5월 9일
0

코딩테스트

목록 보기
412/631
post-thumbnail

1. 문제 설명

주간 미팅

2. 문제 분석

다익스트라 알고리즘을 적게 돌리는 방법은 공통되는 지점을 시작점으로 사용하는 법. (본 문제에서는 장소마다가 아니라 도착점을 시작점으로 사용해 총 두 번만 돌리면 된다)

3. 나의 풀이

import sys
import heapq

INF = sys.maxsize
n, v, e = map(int, sys.stdin.readline().rstrip().split())
a, b = map(int, sys.stdin.readline().rstrip().split())
position = list(map(int, sys.stdin.readline().rstrip().split()))
nodes = [[] for _ in range(v+1)]
for _ in range(e):
    node1, node2, cost = map(int, sys.stdin.readline().rstrip().split())
    nodes[node1].append([node2, cost])
    nodes[node2].append([node1, cost])

def Dijkstra(start):
    distances = [INF for _ in range(v+1)]
    distances[start] = 0
    pq = []
    heapq.heappush(pq, [0, start])

    while pq:
        cur_cost, cur_node = heapq.heappop(pq)
        if distances[cur_node] < cur_cost: continue

        for next_node, next_cost in nodes[cur_node]:
            if distances[next_node] > next_cost + cur_cost:
                distances[next_node] = next_cost + cur_cost
                heapq.heappush(pq, [next_cost + cur_cost, next_node])
    return distances

distances_a = Dijkstra(a)
distances_b = Dijkstra(b)
total = 0

for pos in position:
    dist_a = distances_a[pos]
    dist_b = distances_b[pos]
    if dist_a == INF: total -= 1
    else: total += dist_a
    if dist_b == INF: total -= 1
    else: total += dist_b

print(total)
profile
JUST DO IT

0개의 댓글