[백준 1185] 유럽여행

Junyoung Park·2022년 4월 9일
0

코딩테스트

목록 보기
349/631
post-thumbnail

1. 문제 설명

유럽여행

2. 문제 분석

간선 비용을 재계산해야 원하는 의미의 MST를 구할 수 있다. 노드를 사용하는 비용 + 왕복 비용까지 모두 카운트해야 하기 때문이다.

3. 나의 풀이

import sys
import heapq

def find(node):
    if parents[node] == node: return node
    else:
        parents[node] = find(parents[node])
        return parents[node]
def union(node1,node2):
    root1, root2 = find(node1), find(node2)
    if root1 == root2: return False
    else:
        parents[root2] = root1
        return True
def Kruskal():
    total = 0
    edge_cnt = 0
    while pq:
        cost, node1, node2 = heapq.heappop(pq)
        if union(node1, node2):
            total += cost
            edge_cnt += 1
            if edge_cnt == n-1: return total

n, m = map(int, sys.stdin.readline().rstrip().split())
costs = [0]
for _ in range(n): costs.append(int(sys.stdin.readline().rstrip()))
pq = []
parents = [i for i in range(n+1)]
for _ in range(m):
    a, b, c = map(int, sys.stdin.readline().rstrip().split())
    cost = 2*c + costs[a] + costs[b]
    heapq.heappush(pq, [cost, a, b])
    # 출발 노드 -> 간선 이용 -> 도착 노드 -> 간선 이용: 비용 재계산

answer = min(costs[1:]) + Kruskal()
# 출발 노드는 체류 비용이 가장 적은 노드
print(answer)
profile
JUST DO IT

0개의 댓글