最小生成树kruskal算法(并查集)

时间:2021-01-17 21:32:07

还是图的算法,最小生成树。
kruskal算法和dijkstra算法有异曲同工之处,都有贪心的思想在其中。
先读入边集,将所有的边排序(从小到大),依次加入另一个空集(最小生成树),如果和其中已有的边形成回路,就跳过,看下一条边。用的是并查集判重。最后的解即为最小生成树。

代码如下:

#include <iostream>
#include <cstdio>
#include <algorithm>
#include <cstring>

using namespace std;

int fa[10000],n,i,j,m,ans;

struct node{
int sta;
int fin;
int data;
}set[10000];

int find(int x){//并查集操作
int r=x;
while (fa[r]!=r) r=fa[r];//找父亲
int i=x,t;
while (i!=r){//路径压缩
t=fa[i];
fa[i]=r;
i=t;
}
return r;
}
int main(){
scanf("%d%d",&n,&m);
for (i=1;i<=m;++i){
scanf("%d%d%d",&set[i].sta,&set[i].fin,&set[i].data);
}
for (i=1;i<=m;++i){
fa[i]=i;
}//初始化
for (i=1;i<=m;++i){
if (find(set[i].sta)!=find(set[i].fin)) ans+=set[i].data;
fa[find(set[i].fin)]=set[i].sta; //如果父亲不一样(不在同一个集合,即没有形成回路),就加入生成树中
}
printf("%d",ans);
}