【模板】Kruskal

Kruskal

在此提供模板X1(滑稽护体)

#include<bits/stdc++.h>
using namespace std;
const int N=100010,M=2*N,INF=0x3f3f3f3f;
int n,m;
int p[N];
struct Edge
{
    int a,b,w;
}edges[M];
bool cmp(Edge x,Edge y)
{
    return x.w<y.w;
}
int find(int x)
{
    if(p[x]!=x) p[x]=find(p[x]);
    return p[x];
}
int kruskal()
{
    sort(edges+1,edges+1+m,cmp);
    for(int i=1;i<=n;i++) p[i]=i;
    int res=0,cnt=0;
    for(int i=1;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) return INF;
    return res;
}
int main()
{
    cin>>n>>m;
    for(int i=1;i<=m;i++)
    {
        int a,b,w;
        cin>>a>>b>>w;
        edges[i]={a,b,w};
    }
    int t=kruskal();
    if(t==INF) cout<<"impossible";
    else cout<<t;
    return 0;
}