Kruskal算法求最小生成树

例题链接

Kruskal算法的对比prim的区别是，前者使用了并查集的思路，而后者类似于Dijkskal算法。操作策略是，对所有边权进行从小到大的排序，然后从小到大遍历所有边权去选择，一个边被选择的条件是，这条边的两个端点位于不同的集合，即二者的find函数得到的值不同，然后将两个点加到同一个集合里面即可，最后选择n - 1条边就是最小生成树

#include<algorithm>
#include<iostream>
#include<cstdio>
using namespace std;
const int N = 200010;
int p[N];
int n,m;
struct Edge//边权结构体
{
    int a,b,w;
    bool operator< (const Edge &W)const//重载小于号，用于后面sort排序
    {
        return w < W.w;
    }
}edges[N];
int find(int x)//并查集模板
{
    if(p[x] != x) p[x] = find(p[x]);
    return p[x];
}
int main()
{
    scanf("%d%d",&n,&m);
    for(int i = 0;i < m;i ++)
    {
        int a,b,w;
        scanf("%d%d%d",&a,&b,&w);
        edges[i] = {a,b,w};
    }
    sort(edges, edges + m);//对边权进行从小到的排序
    for(int i = 1;i <= n;i ++) p[i] = i;
    int res = 0,cnt = 0;
    for(int i = 0;i < m;i ++)//从边权小的开始选取，如果两个点在同一个集合里这个边就不能选择，否则就可以进行选择
    {
        int a = edges[i].a,b = edges[i].b,w = edges[i].w;
        a = find(a),b = find(b);
        if(a != b)//不在同一个集合里的操作
        {
            p[a] = b;
            res += w;
            cnt ++; 
        }
    }
    if(cnt < n - 1) puts("impossible");//最后如果选取的边数不是n - 1则无法构成一棵生成树
    else printf("%d\n",res);
    return 0;
}

标签：const,int,Kruskal,d%,最小,find,算法,edges,边权
来源： https://www.cnblogs.com/amour233/p/16483891.html