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

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

代码如下:

#include 
#include 
#include 
#include 

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);
}

你可能感兴趣的:(图)