算法策略:边贪心。
容易理解。
并查集可以通过查询两个结点所在集合的根节点是否相同来判断他们是否在同一个集合(即是否连通)。
int kruskal()
{
令最小生成树的边权值之和为ans,最小生成树的当前边数为Num_Edge
将所有边按边权从小到大排序;
for(枚举所有边)
{
if 当前测试的两个端点在不同的连通块中
将该测试边加入最小生成树中;
ans += 测试边的边权;
最终生成树的当前边数NumEdge 加1;
当前边数Num_Edge=顶点数-1时结束循环;
}
return ans;
}
struct edge{
int u,v; //边的两个端点
int cost;//边权
};
bool cmp(edge a,edge b){
return a.cost<b.cost;
}
int father[N];//并查集数组
int findFather(int x){//并查集查询函数,找最大的老大
int a = x;
while(x!=father[x]){
x = father[x];
}//x就是找到的最大的老大
//并查集路径压缩,将所有的小弟的老大都变成最大的老大
while(a!=father[a]){
int z = a;
a = father[a];
father[z] = x;
}
return x;
}
//kruskal函数返回最小生成树的边权之和,参数n为顶点个数,m为图的边数
int kruskal(int n,int m){
//ans为所求边权值和,num_Edge为当前生成树的边数
int ans = 0,Num_Edge = 0;
for(int i = 0;i < n;i++){//假设题目中顶点范围是[1,n]
father[i] = i;//并查集初始化
}
sort(E,E+m,cmp) ;//所有边按边权从小到大排序
for(int i=0;i<m;i++)//枚举所有边
{
int faU = findFather(E[i].u);//查询测试两个端点所在集合的根节点
int faV = finidFather(E[i].v);
if(faU != faV){//如果不在一个集合中
father[faU] = faV;//合并集合 (即把测试加入最小生成树中)
ans += E[i].cost;//边权之和增加测试边的边权
Num_Edge++;//当前生成树的边数加一
if(Num_Edge == n-1) break;//边数等于顶点数减一时结束算法
}
}
if(Num_Edge != n-1) return -1;//无法连通时返回-1
else return ans; //返回最小生成树边权之和
}