본문 바로가기
Problem Solving/Union-Find & Minimum Spanning Tree

[ 최소 신장 트리 ] 백준 1922번: 네트워크 연결 / 골드 4

by ggyongi 2021. 7. 29.
반응형

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

 

1922번: 네트워크 연결

이 경우에 1-3, 2-3, 3-4, 4-5, 4-6을 연결하면 주어진 output이 나오게 된다.

www.acmicpc.net

def find(parent, x):
    if parent[x] != x:
        parent[x] = find(parent, parent[x])
    return parent[x]

def union(parent, a, b):
    a = find(parent, a)
    b = find(parent, b)
    if a < b:
        parent[b] = a
    else:
        parent[a] = b

import sys
n = int(input())
m = int(input())
parent = [0]*(n+1)
edges = []
result = 0

for i in range(n+1):
    parent[i] = i

for _ in range(m):
    a, b, cost = map(int, sys.stdin.readline().split())
    edges.append([cost, a, b])

edges.sort()

for i in range(len(edges)):
    cost, a, b = edges[i]
    if find(parent, a) != find(parent, b):
        union(parent, a, b)
        result += cost

print(result)
 

비전공자 네카라 신입 취업 노하우

시행착오 끝에 얻어낸 취업 노하우가 모두 담긴 전자책!

kmong.com

댓글