最小生成树 kruscal算法(涉及并查集的应用)

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

using namespace std;

const int N=100010,M=200010,INF=0x3f3f3f3f;

int p[N];int n,m;int cnt=0,res=0;

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 kruscal()
{
    sort(edges,edges+m);
    
    for(int i=1;i<=n;i++) p[i]=i;//初始化并查集
    
    for(int i=0;i<m;i++)
    {
        int a=edges[i].a, b=edges[i].b, w=edges[i].w;
        if(find(a)!=find(b))
        {
            p[find(a)]=find(b);
            res+=w;
            cnt++;
        }
    }
    if(cnt < n-1) return INF;
    return res;
}

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};
    }
    
    int t=kruscal();
    
    if(t==INF) puts("impossible");
    else 
       printf("%d\n",t);
       
    return 0;
}

 

posted @ 2023-11-06 20:12  敲代码的6  阅读(28)  评论(0)    收藏  举报