HDOJ 1233 还是畅通工程(并查集)



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

还是畅通工程

Time Limit: 4000/2000 MS (Java/Others)    Memory Limit: 65536/32768 K (Java/Others)
Total Submission(s): 25902    Accepted Submission(s): 11536

Problem Description

某省调查乡村交通状况，得到的统计表中列出了任意两村庄间的距离。省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可），并要求铺设的公路总长度为最小。请计算最小的公路总长度。

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( < 100 )；随后的N(N-1)/2行对应村庄间的距离，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间的距离。为简单起见，村庄从1到N编号。
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最小的公路总长度。

 

Sample Input

  
  

   
   3
1 2 1
1 3 2
2 3 4
4
1 2 1
1 3 4
1 4 1
2 3 3
2 4 2
3 4 5
0
  
  

 

Sample Output

  
  

   
   3
5


   
   

    
    

     
     Hint
    
    
Hint
   
   
   
    
Huge input, scanf is recommended.
  
  
  
  


  
  
  
  

   
   第一次写时WA找不到错误，第二次在第一次写的基础上改变了判断两村庄在前边的合并下已经在一个集合中的代码(第一次代码的67~83行)，然后过了，这样看来应该是这里的错，但始终不知道为什么，大神指教。
  
  
  
  

   
   先贴第二次AC的代码，代码后边附带几组测试数据。
  
  
  
  

   
   
   
   
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int set[10000];
struct Road
{
	int NUM_1;
	int NUM_2;
	int distance;
}roads[10000];

int cmp(const void *a,const void *b)
{
	return ((Road *)a)->distance - ((Road *)b)->distance;
}
int Find(int x)
{
	int r=x,i=x,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 T,N,i,j,k,S,sum;
	while(scanf("%d",&T)&&T!=0)
	{
		memset(set,0,sizeof(set));
		N=T*(T-1)/2;
		set[0]=-1;//后边要排序，用不到set[0]，所以给他赋-1让他始终在第一个，不影响后边的数 
		for(i=1;i<=T;i++)//初始化，每个村庄都是独立的，即每个数的根就是自己 
		set[i]=i;
		for(i=1;i<=N;i++)
		scanf("%d %d %d",&roads[i].NUM_1,&roads[i].NUM_2,&roads[i].distance);
		qsort(roads,N+1,sizeof(roads[0]),cmp);///按道路从小到大排序结构体 
		
		for(i=1,sum=0;i<=N;i++)///到i组数据时，全部的村庄连通 
		{
			  if(Find(roads[i].NUM_1)!=Find(roads[i].NUM_2))//判断村庄NUM_1和NUM_2是否有相同的根 
			  sum+=roads[i].distance;//若没有把NUM_1和NUM_2之间的卢修建起来 
		      merge(roads[i].NUM_1,roads[i].NUM_2);//再把NUM_1和NUM_2归于一个根(合并到一个集合里) 
		      
			  for(j=1,S=0;j<=T;j++)//所有的村庄是否连通 
			  if(set[j]==j)
			  S+=1;
			  
			  if(S==1)//连通则跳出 
			  break;
		}
		printf("%d\n",sum);
	}
	return 0;
}
/*
5
1 2 5
1 3 5
1 4 5
1 5 5
2 3 4
2 4 4
2 5 4
3 4 3
3 5 3
4 5 2

6
1 2 6
1 3 6
1 4 6
1 5 6
1 6 6
2 3 5
2 4 5
2 5 5
2 6 5
3 4 4
3 5 4
3 6 4
4 5 3
4 6 3
5 6 2

14
20
*/
   
   

  
  
  
  

   
   下面是第一次WA的代码
  
  
  
  

   
   
   
   
//此程序能想到的测试数据都对，但就是WA。
//改了一下这个程序的判断村庄是否出现代码(67~83行),改成在合并时就判断。但感觉现在的也没错，求大神指教。 
#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int set[10000];
struct Road
{
	int NUM_1;
	int NUM_2;
	int distance;
}roads[10000];

int cmp(const void *a,const void *b)
{
	return ((Road *)a)->distance - ((Road *)b)->distance;
}
int Find(int x)
{
	int r=x,i=x,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 T,N,i,j,k,S,sum;
	while(scanf("%d",&T)&&T!=0)
	{
		memset(set,0,sizeof(set));
		N=T*(T-1)/2;
		set[0]=-1;//后边要排序，用不到set[0]，所以给他赋-1让他始终在第一个，不影响后边的数 
		for(i=1;i<=T;i++)//初始化，每个村庄都是独立的，即每个数的根就是自己 
		set[i]=i;
		for(i=1;i<=N;i++)
		scanf("%d %d %d",&roads[i].NUM_1,&roads[i].NUM_2,&roads[i].distance);
		qsort(roads,N+1,sizeof(roads[0]),cmp);///按道路从小到大排序结构体 

		
		for(i=1;i<=N;i++)///到i组数据时，全部的村庄连通 
		{
		      merge(roads[i].NUM_1,roads[i].NUM_2);
			  for(j=1,S=0;j<=T;j++)
			  if(set[j]==j)
			  S+=1;
			  
			  if(S==1)
			  break;
		}
		if(N==0/*也就是T==1*/) i=0;
		for(j=1,sum=0;j<=i;j++)
		{
			int judge_1=0,judge_2=0;//判断前边的道路是否已经把两个村庄已经连通
		    //judge_1==0,judge_2==0分别表示没有连通    
		    for(k=1;k<j;k++)
			{
			     if(roads[k].NUM_1==roads[j].NUM_1||roads[k].NUM_2==roads[j].NUM_1)
				 judge_1=1;
				 if(roads[k].NUM_1==roads[j].NUM_2||roads[k].NUM_2==roads[j].NUM_2)
				 judge_2=1;
			}
			
			if(judge_1==1&&judge_2==1)
			continue;
			else
			sum+=roads[j].distance;
        } 
		
		printf("%d\n",sum);
	}
	return 0;
}