HDOJ -- 1863畅通工程

畅通工程

Time Limit:1000MS     Memory Limit:32768KB     64bit IO Format:%I64d & %I64u

Description

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

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
?

---

解题思路：这道题跟HDOJ1233类似，在原基础上改改就行了。但题目还点儿特殊的地方，就是当条件不足时要输出“？”。原本的思路是按照n的特殊情况讨论，比如第二个输入样例当n<m-1时，无论如何也不可能实现全省畅通。可要是即便n>=m-1，要是给出的数据中有村庄重复的怎么办？？好吧！这个思路被扼杀了。。

参考了（还是参考。。）大神的代码，突然明白了可以在最后判断树的棵数，如果就只有一棵树，那么肯定满足了条件。如果树的棵数大于一，那么就说明还有村庄未联通！树的棵数不就是根节点的个数嘛。。。

#include<stdio.h>
#include<string.h>
#include<algorithm>
using namespace std;
int per[5000],n,m;
struct node{
    int a,b;//道路两端的村庄 
    int dis;//道路之间的成本 
}x[5005];
bool cmp(node a,node b){
    return a.dis<b.dis;
}
void init(){
	for(int i=1;i<=m;++i)
       	per[i]=i;
}
int find(int x){
    int r=x;
    while(r!=per[r])
        r=per[r];
    int i=x,j;
    while(i!=r){
        j=per[i];
        per[i]=r;
        i=j;
    }
    return r;
}//并查集老套路，路径压缩 
bool join(int a,int b){
    int fx=find(a);
	int fy=find(b);
    if(fx!=fy){
        per[fx]=fy;
        return 1;
    }
    return 0;
}
int main(){
    while(scanf("%d%d",&n,&m),n){
    	init();
        for(int i=0;i<n;i++)
            scanf("%d%d%d",&x[i].a,&x[i].b,&x[i].dis);
        sort(x,x+n,cmp); 
        int sum=0;
        for(int i=0;i<n;i++){
            if(join(x[i].a,x[i].b)){
                sum+=x[i].dis;
            }
        }
		int cnt=0;
        for(int i=1;i<=m;i++){//如果全省畅通，则所有村庄都连接在一个根节点上 
        	if(per[i]==i)//当i是根节点时 
        		cnt++;
		}
		if(cnt>1)//如果根节点数大于1，说明还存在未联通的村庄 
			printf("?\n");
		else
        	printf("%d\n",sum);
    }
    return 0;
}