[백준] 1197번 : 최소 스패닝 트리 (python 파이썬)

에딕·2021년 3월 12일
0

백준

목록 보기
8/13
post-thumbnail

👉 1197번 : 최소 스패닝 트리



✍ 내 코드


# 골드 4레벨        최소 스패닝 트리
# 크루스칼 알고리즘
# 유니온 파인드
from sys import stdin

read = stdin.readline
V, S = map(int, read().split())

edge = []
for _ in range(S):
    a, b, w = map(int, read().split())
    edge.append((w, a, b))

edge.sort(key=lambda x: x[0])

parent = list(range(V + 1))


def union(a, b):
    a = find(a)
    b = find(b)

    if b < a:
        parent[a] = b
    else:
        parent[b] = a


def find(a):
    if a == parent[a]:
        return a
    parent[a] = find(parent[a])  # 경로 압축
    return parent[a]


sum = 0
for w, s, e in edge:
    if find(s) != find(e):
        union(s, e)
        sum += w

print(sum)


✍ 팁


유니온 파인드에서 실수를 자각을 못해서 오래 걸렸다...ㅠㅠ

  • 크루스칼 알고리즘을 알면 문제 없이 풀리는 문제
profile
코딩💻 고양이😺

0개의 댓글