AcWing 859. Kruskal算法求最小生成树

kruskal是基于边的算法，所以我们边用一个结构体来存就可以了。列举每条边的时候，用到了前面并查集的知识。由于需要对边集进行排序，所以我们记得在边结构体定义的时候重载一下<

并查集：

837. 连通块中点的数量 - AcWing题库

原题链接如下：

859. Kruskal算法求最小生成树 - AcWing题库

#include<iostream>
#include<algorithm>
using namespace std;
const int N = 100010;
const int M = 200010;
const int INF = 0x3f3f3f3f;
int p[N];
int n,m;

struct Edge
{
    int a,b,w;
    
    //重载一下小于号
    bool operator< (const Edge &W)const
    {
        return w<W.w;
    }
}edges[M];

int find(int x)
{
    //并查集中的模板，查找是否某一个点的祖宗节点编号。
    if(p[x]!=x) p[x] = find(p[x]);
    return p[x];
}

int kruskal()
{
    //1.先将边集排序
    sort(edges,edges+m);
    //2.初始化并查集
    for(int i = 0;i<n;i++)p[i] = i;
    
    //开始枚举每条边
    int res = 0,cnt = 0;
    for(int i = 0;i<m;i++)
    {
        int a = edges[i].a;
        int b = edges[i].b;
        int w = edges[i].w;
        
        a = find(a),b = find(b);//这里是并查集模板
        if(a!=b)
        {
            p[a] = b;
            res+=w;
            cnt++;
        }
    }
    
    if(cnt<n-1)return INF;
    return res;
}
int main()
{
    cin>>n>>m;
    for(int i = 0;i<m;i++)
    {
        int a,b,w;
        cin>>a>>b>>w;
        edges[i] = {a,b,w};
    }
    
    int t = kruskal();
    
    if(t == INF)puts("impossible");
    else cout<<t<<endl;
    
    return 0;
}