最短路径-畅通工程

畅通工程

  HDU - 1863 

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。经过调查评估，得到的统计表中列出了有可能建设公路的若干条道路的成本。现请你编写程序，计算出全省畅通需要的最低成本。 

Input测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N 
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。 
Output对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。 
Sample Input

3 3
1 2 1
1 3 2
2 3 4
1 3
2 3 2
0 100

Sample Output

3
?

#include <iostream>
#include <algorithm>
using namespace std;
#include <cstdio>
#include <cstdlib>
#include <cstring>
const int maxm = 101;
const int maxn = 5055;
int N,M;
int edgenum = 0;
int f[maxm];
struct node
{
    int from,to,w;
    node(int a,int b,int c)
    {
        from = a;
        to = b;
        w = c;
    }
}edge[maxn];
bool cmp(node a, node b)
{
    return a.w < b.w;
}
int getfather(int i)
{
    if(f[i] == i) return i;
    else return f[i] = getfather(f[i]);
}
int main()
{
    while(scanf("%d%d",&N,&M) && N)
    {
        edgenum = 0;
        for(int i = 0; i < maxm; i++) f[i] = i;
        for(int i = 0; i < N; i++)
        {
            int a,b,c;
            scanf("%d%d%d",&a,&b,&c);
            edge[edgenum++] = node(a,b,c);
        }
        sort(edge,edge+edgenum,cmp);//按照cmp的规则排序
        int num = 0,ans = 0;
        bool  judge = false;
        for(int i = 0; i < edgenum; i++)
        {
            int a = getfather(edge[i].from);
            int b = getfather(edge[i].to);
            if(a != b)
            {
                f[a] = b;
                num ++;
                ans += edge[i].w;
                if(num == M-1)
                {judge =  true;break;}
            }
        }
        if(!judge) printf("?\n");
        else printf("%d\n",ans);
    }
    return 0;
}