[6] template minimum spanning tree (Kruskal)

 

 

 

 

 

 

#include<iostream>
#include<cstdio>
#include<string.h>
#include<algorithm>
using namespace std;
struct p
{
    int x;
    int y;
    int t;
}zyf[200001];
bool cmp(p xx,p yy)
{
    return xx.t<yy.t;
}
int total=0,sum=0;
int fa[5001],z,n,m,a,b,k;
int getfather(int w)
{
    if(fa[w]==w)
    return w;
    fa[w]=getfather(fa[w]);
    return fa[w];
}
void together(int u,int o)
{
    fa[getfather(u)]=getfather(o);
}
int main()
{
    //freopen("testdata (23).in","r",stdin);
    cin>>n>>m;
    for(int i=1;i<=n;i++)
    {
        fa[i]=i;
    }
    for(int i=1;i<=m;i++)
    {
        cin>>zyf[i].x>>zyf[i].y>>zyf[i].t;
    }
    sort(zyf+1,zyf+m+1,cmp); 
    for(int i=1;i<=m;i++)
    {
        if(getfather(zyf[i].x)!=getfather(zyf[i].y))
        {
            together(zyf[i].x,zyf[i].y);
            sum+=zyf[i].t;
            for(int j=1;j<=n;j++)
            {
                int cc=getfather(j);
                if(cc==j)
                total++;
            }
            if(total==1)
            break;
            total=0;
        }
    }
    if(total==1)
    cout<<sum;
    else
    {
        cout<<"orz";
    }
    return 0;
} 

                        2019-09-09 By the fourteenth day of summer

Guess you like

Origin www.cnblogs.com/north-star/p/11489423.html