HDOJ 1863 畅通工程(并查集)



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

畅通工程

Time Limit: 1000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 16976    Accepted Submission(s): 7120

Problem 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>
#include<string.h>
#include<stdlib.h>
int set[1000];
struct Road//P为NUM_1与NUM_2之间道路的费用 
{
	int NUM_1;
	int NUM_2;
	int P;
}roads[1000];
int cmp(const void *a,const void *b)//按道路费用从低到高排练结构体 
{
	return ((Road *)a)->P - ((Road *)b)->P;
}
int Find(int x)
{
	int r=x,i=r,j;
	while(set[r]!=r)
	r=set[r];
	
	while(i!=r)
	{
		j=set[i];
		set[i]=r;
		i=j;
	}
	return r;
}
void merge(int x,int y)
{
	x=Find(x);
	y=Find(y);
	if(y<x)
	set[x]=y;
	if(x<y)
	set[y]=x;
}
int main()
{
	int N,M,i,j,S,sum_p;
	while(scanf("%d %d",&N,&M)&&N!=0)
	{
		memset(set,0,sizeof(set));
		for(i=1;i<=M;i++)
		set[i]=i;
		roads[0].P=-1;//数组为0的赋-1，排序后在第一位，不影响后边的排序 
		for(i=1;i<=N;i++)
		scanf("%d %d %d",&roads[i].NUM_1,&roads[i].NUM_2,&roads[i].P);
		
		qsort(roads,N+1,sizeof(roads[0]),cmp);//注意排序的数目为N+1 
		
		for(i=1;i<=N;i++)//每用merge()合并一次就查看一遍是否所有个村庄是否有一个根(即是否在一个集合中) 
		{
			 S=0;
		     merge(roads[i].NUM_1,roads[i].NUM_2);
			 for(j=1;j<=M;j++)
			 {
			 if(set[j]==j)
			 S+=1;
 			//printf("%d %d#\n",j,set[j]);
		     }
			 if(S==1)//如果所有村庄有同一个根，就跳出，费用加到roads[i].P 
			 break;
		}
		for(j=1,sum_p=0;j<=i;j++)
		sum_p+=roads[j].P;
		
		if(S>1)
		printf("?\n");//如果所有的村庄始终都不在一个集合中，就不满足连通 
		else
		printf("%d\n",sum_p);
		
	}
	return 0;
}