hdu 2122 Ice_cream’s world III（最小生成树）

#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<algorithm>
#include<queue>
#include<stack>
#define mem(a,b) memset(a,b,sizeof(a))
#define ll __int64
#define MAXN 1000
#define INF 0x7ffffff
#define lson l,m,rt<<1
#define rson m+1,r,rt<<1|1
using namespace std;
struct Road
{
    int u,v,w;
};
Road r[10000+10];
int fat[1000+10];
int n,m,sum,ans;
int cmp(Road a,Road b)
{
    return a.w<b.w;
}
int find(int x)
{
    return fat[x]==x?x:find(fat[x]);
}
void Kru()
{
    sum=1;ans=0;
    for(int i=0;i<m;i++)
    {
        int x=find(r[i].u);
        int y=find(r[i].v);
        if(x!=y)
        {
            ans+=r[i].w;
            sum++;
            fat[x]=y;
        }
    }
    if(sum==n)    
        printf("%d\n\n",ans);
    
    else    
        printf("impossible\n\n");          
}
int main()
{
    int i,j;    
    while(scanf("%d%d",&n,&m)!=EOF)
    {
        for(i=0;i<n;i++)        
            fat[i]=i;        
        for(i=0;i<m;i++)
            scanf("%d%d%d",&r[i].u,&r[i].v,&r[i].w);                   
        sort(r,r+m,cmp);
        Kru();
    }
    return 0;
}