还是畅通工程

畅通工程

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

Submit  Status

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
?

注意统计能联通的村庄数


#include<stdio.h>//d
#include<string.h>
#include<algorithm>
#include<iostream>
using namespace std;
const int M = 9999999;
int flag;
struct node{
	int from,to,val;
};
int f[M];
node dis[M];
char vis[M];

bool comparison(node a,node b){  
	return a.val<b.val;  
}

int find(int x){
	return x == f[x]?x:f[x] = find(f[x]);
}

void Union(int a,int b)  
{  
    int a1=find(a);  
    int b1=find(b);  
    if(a1!=b1)  
        f[a1]=b1;
		flag++;  
}  


int main(){
	int n,m,sum;
	while(scanf("%d%d",&n,&m)!=EOF){
		if(n == 0)	break;
		sum = 0;flag = 1;
		for(int i=0; i<=n; i++)	f[i] = i;
		memset(vis,0,sizeof(vis));
		for(int i=0; i<n; i++){
			
			scanf("%d%d%d",&dis[i].from ,&dis[i].to ,&dis[i].val );
		}
		/*if(n == 1)
		{
			printf("0\n");	continue;
		}*/
		sort(dis,dis+n,comparison); 
		for(int i=0; i<n; i++){
			if(find(dis[i].from)!= find(dis[i].to))
			{
			Union(dis[i].from ,dis[i].to );
				sum += dis[i].val ; 
			}
		}
		if(flag == m)
			printf("%d\n",sum);
		else
			printf("?\n");
		
		
	}
	return 0;
}