畅通工程

http://acm.hdu.edu.cn/showproblem.php?pid=1863

#include<cstring>
#include<iostream>
#include<cstdio>
#include<algorithm>
using namespace std;
struct road
{
    int x,y,l;
}ht[10010];
int set[105];
int find(int x)
{
    if(set[x]!=x)
    set[x]=find(set[x]);
    return set[x];
}
int merge(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y)
    return 0;
    set[x]=y;
    return 1;
}
int cmp(road a,road b)
{
    return a.l<b.l;
}
int main()
{
    int m,n,i,num,k;
    while(~scanf("%d%d",&m,&n))
    {
       if(m==0)
       break;
       for(i=1;i<=n;i++)
       set[i]=i;
       for(i=0;i<m;i++)
       {
           scanf("%d%d%d",&ht[i].x,&ht[i].y,&ht[i].l);
       }
       if(m<n-1)
       {
           printf("?\n");
           continue;
       }
       sort(ht,ht+m,cmp);
        num=0;
        k=1;
        i=0;
        while(k<n)
        {
            if(i>=m)
            break;
            if(merge(ht[i].x,ht[i].y))
            {
                num+=ht[i].l;
                k++;
            }
            i++;
        }
        if(k<n)
        printf("?\n");
        else
        printf("%d\n", num) ;
    }
    return 0;
}