简述:
Kruskal算法按照边的权值的顺序从小到大查看一遍,如果不产生圈(重边等也算在内),就把当前这条边加入到生成树中。至于这个算法为什么是正确的,其实和Prim算法证明的思路基本相同,在此就不详细说明了。
接下来我们介绍如何判断是否产生圈。假设现在要把连接顶点u和顶点v的边e加入生成树中。如果加入之前u和v不在同一个连通分量里,那么加入e也不会产生圈。反之,如果u和v在同一个连通分量里,那么一定会产生圈。可以使用并查集高效地判断是否属于同-一个连通分量。
Kruskal算法在边的排序上最费时,算法的复杂度是O(E*logV)。
实现:
#include<iostream>
#include<vector>
#include<algorithm>
using namespace std;
int n,m;
struct edge {
int from,to,val;
bool operator<(const struct edge a) const {
return val<a.val;
}
};
vector<struct edge>mp;
vector<int>father;
void init() {
father.resize(n);0
for(int i=0; i<n; ++i)
father[i]=i;
}
int find(int x) {
return x==father[x]?x:father[x]=find(father[x]);
}
bool same(int x,int y) {
return find(x)==find(y);
}
void Union(int x,int y) {
father[father[x]]=father[y];
}
int Kruskal() {
init();
sort(mp.begin(),mp.end());
int res=0;
for(int i=0; i<mp.size(); ++i) {
if(!same(mp[i].from,mp[i].to)) {//选择最小的边合并顶点
Union(mp[i].from,mp[i].to);
res+=mp[i].val;
}
}
return res;
}
int main() {
cin>>n>>m;
for(int i=0; i<m; ++i) {
int a,b,v;
cin>>a>>b>>v;
mp.push_back({a,b,v});
}
cout<<Kruskal();
return 0;
}