Python使用并查集的思想求解最小生成树问题 输入格式: n,e ;n端点数量、m边的数量 n1,n2,e1 ; n1端点1,n2端点2 e1边的权值 。。。 ;这样的信息有e行 输出格式: n1 n2 e1 生成树中的边的信息
时间: 2024-02-05 12:14:14 浏览: 88
以下是Python实现并查集的思想求解最小生成树问题的代码:
```python
class UnionFind:
def __init__(self, n):
self.parent = list(range(n))
self.rank = [0] * n
def find(self, x):
if self.parent[x] != x:
self.parent[x] = self.find(self.parent[x])
return self.parent[x]
def union(self, x, y):
px, py = self.find(x), self.find(y)
if px == py:
return False
if self.rank[px] > self.rank[py]:
self.parent[py] = px
elif self.rank[px] < self.rank[py]:
self.parent[px] = py
else:
self.parent[py] = px
self.rank[px] += 1
return True
def kruskal(n, edges):
uf = UnionFind(n)
edges.sort(key=lambda x: x[2])
res = []
for u, v, w in edges:
if uf.union(u, v):
res.append((u, v, w))
return res
n, m = map(int, input().split())
edges = []
for _ in range(m):
u, v, w = map(int, input().split())
edges.append((u-1, v-1, w))
res = kruskal(n, edges)
for u, v, w in res:
print(u+1, v+1, w)
```
输入格式为第一行两个整数n和m,表示端点数量和边的数量。接下来m行每行三个整数u、v、w,表示一条边连接u和v两个端点,边权为w。输出格式为生成树中的边的信息,每行三个整数u、v、w,表示生成树中连接u和v两个端点的边权为w的边。
阅读全文