畅通工程(并查集模版题)

#include<cstdio>
#include<algorithm>
#include<iostream>
#include<string.h>
using namespace std;
int fa[105];
struct node//定义这个结构体，a代表的是一个村庄的编号，b代表的是另一个村庄的编号，cost顾名思义 
{
    int a;
    int b;
    int cost;    
};
int find(int x)
{
    int r=x,temp;
    while(r!=fa[r])
    r=fa[r];
    while(x!=fa[x])
    {
        temp=fa[x];
        fa[x]=r;
        x=temp;
    }
    return x;
}
bool cmp(node x,node y)//cmp排序 
{
    return x.cost<y.cost;
}
int main(void)
{
    int n,m;
    node cp[500];
    while(cin>>n>>m&&n)
    {
        int sum=0;
        for(int i=1;i<=m;i++)
        fa[i]=i;
        for(int i=1;i<=n;i++)
        {
            cin>>cp[i].a>>cp[i].b>>cp[i].cost;
        }
        sort(cp+1,cp+n+1,cmp);//排序 
        for(int i=1;i<=n;i++)//这个循环开始处理费用 
        {
            int x=find(cp[i].a);
            int y=find(cp[i].b);
            if(x!=y)
            {
                fa[y]=x;
                sum+=cp[i].cost;
            }
        }
        int k=find(1);
        int flag=1;
        for(int i=1;i<=m;i++)//这个循环检查是否村子都连接 
        {
            if(k!=find(i))
            {
                flag=0;
                break;
            }
    }
    if(flag)
    cout<<sum<<endl;
    else
    cout<<‘?‘<<endl;
}
return 0;
}