[백준 2406] 안정적인 네트워크

Junyoung Park·2022년 3월 20일
0

코딩테스트

목록 보기
288/631
post-thumbnail

1. 문제 설명

안정적인 네트워크

2. 문제 분석

1번 노드를 제외한 MST를 만든다. 문제 해석이 관건.

3. 나의 풀이

import sys
import heapq

n, m = map(int, sys.stdin.readline().rstrip().split())
pq = []
parents = [i for i in range(n+1)]

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

for _ in range(m):
    a, b = map(int, sys.stdin.readline().rstrip().split())
    union(a, b)
for i in range(1, n+1):
    line = list(map(int, sys.stdin.readline().rstrip().split()))
    if i == 1: continue
    # 본사 컴퓨터 1번 노드를 제외한 MST를 만들기 위함
    for j in range(i+1, n+1):
        heapq.heappush(pq, [line[j-1], i, j])

MST = 0
edge_cnt = 0
result = []
while pq:
    cur_cost, node1, node2 = heapq.heappop(pq)
    if union(node1, node2):
        MST += cur_cost
        edge_cnt += 1
        result.append([node1, node2])
        if edge_cnt == n-1: break
    # 크루스칼 알고리즘

print(MST, len(result))
for node1, node2 in result:
    print(node1, node2)
profile
JUST DO IT

0개의 댓글