🔔 신장 트리
🔔 Kruskal 알고리즘의 동작
1. 그래프의 간선들을 가중치의 오름차순으로 정렬
2. 정렬된 간선 리스트에서 순서대로 사이클을 형성하지 않는 간선을 선택함
- 가장 낮은 가중치를 먼저 선택
- 사이클을 형성하는 간선을 제외
3. 해당 간선을 현재의 MST(최소 비용 신장 트리) 집합에 추가함
🔔 Kruskal 알고리즘 구현
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
def kruskal(graph):
edges = []
for node in graph:
for neighbor, weight in graph[node]:
edges.append((weight, node, neighbor))
edges.sort()
parent = [i for i in range(len(graph))]
mst = []
for edge in edges:
weight, a, b = edge
if find_parent(parent, a) != find_parent(parent, b):
union_parent(parent, a, b)
mst.append(edge)
return mst
⭐ 섬 연결하기