- 최소 신장 트리 알고리즘 (사이클이 없음)
- edge의 개수 = vertex의 개수 - 1
- 시간 복잡도 :
O(E*logE)
, E개의 간선 정렬
- 간선 데이터를 비용에 따라 오름차순으로 정렬한다.
- 간선을 하나씩 확인하며 현재 간선이 사이클을 발생시키는지 확인한다.
2-1. 사이클이 발생하지 않는 경우에만 최소 신장 트리에 포함시킨다.
- 모든 간선에 대하여 2를 반복한다.
def find_parent(parent, x):
if parent[x] != x:
parent[x] = find_parent(parent, parent[x])
return parent[x]
def union_parent(parent, a, b):
a = find_parent(parent, a)
b = find_parent(parent, b)
if a < b:
parent[b] = a
else:
parent[a] = b
v, e = map(int, input().split())
parent = []
for i in range(v + 1):
parent.append(i)
edges = []
for _ in range(e):
a, b, cost = map(int, input().split())
edges.append((cost, a, b))
edges.sort()
result = 0
for edge in edges:
cost, a, b = edge
if find_parent(parent, a) != find_parent(parent, b):
union_parent(parent, a, b)
result += cost
print(result)
'''
7 9
1 2 29
1 5 75
2 3 35
2 6 34
3 4 7
4 6 23
4 7 13
5 6 53
6 7 25
'''
'''
159
'''