HDOJ 1879 继续畅通工程(并查集)

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

继续畅通工程

Time Limit: 2000/1000 MS (Java/Others)    Memory Limit: 32768/32768 K (Java/Others)
Total Submission(s): 13805    Accepted Submission(s): 6015

Problem Description

省政府“畅通工程”的目标是使全省任何两个村庄间都可以实现公路交通（但不一定有直接的公路相连，只要能间接通过公路可达即可）。现得到城镇道路统计表，表中列出了任意两城镇间修建道路的费用，以及该道路是否已经修通的状态。现请你编写程序，计算出全省畅通需要的最低成本。

 

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出村庄数目N ( 1< N < 100 )；随后的 N(N-1)/2 行对应村庄间道路的成本及修建状态，每行给4个正整数，分别是两个村庄的编号（从1编号到N），此两村庄间道路的成本，以及修建状态：1表示已建，0表示未建。

当N为0时输入结束。

 

 

Output

每个测试用例的输出占一行，输出全省畅通需要的最低成本。

 

 

Sample Input

  
  

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

 

 

Sample Output

  
  

   
   3
1
0
  
  

 

#include<stdio.h>
#include<string.h>
#include<stdlib.h>
int set[10000];
struct Road
{
     int NUM_1;
     int NUM_2;
     int P;
     int link;
}roads[10000];
int cmp(const void *a,const void *b)
{
    struct Road *c=(Road *)a;
    struct Road *d=(Road *)b;
    if(c->link!=d->link)
    return d->link - c->link;
    else
    return c->P - d->P;
}
int Find(int x)
{
    while(set[x]!=x)
    x=set[x];
    return x;
}
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,k,S,sum;
    while(scanf("%d",&N)&&N!=0)
    {
            M=N*(N-1)/2;      
                        
            memset(set,0,sizeof(set));
            for(i=1;i<=N;i++)
            set[i]=i;
            roads[0].link=2;//用不到，始终排到第一个 
            for(i=1;i<=M;i++)                     
            scanf("%d %d %d %d",&roads[i].NUM_1,&roads[i].NUM_2,&roads[i].P,&roads[i].link);
            qsort(roads,M+1,sizeof(roads[0]),cmp);
            
            for(i=M;i>=1;i--)
            if(roads[i].link!=0) break;
            for(j=1;j<=i;j++)//将有路的村庄连通 
            merge(roads[j].NUM_1,roads[j].NUM_2);
            for(j=1,S=0;j<=N;j++)//判断是否全部连通 
            if(set[j]==j)
            S+=1;
            
            if(S==1)//已全部连通
            printf("0\n");
            else
            {
                for(j=i+1,sum=0;j<=M;j++)
                {//从没连通的数据i+1开始判断 
                       if(Find(roads[j].NUM_1)!=Find(roads[j].NUM_2))
                       {//如果j组数据中两个村庄没有连通，那么就要修这条路，然后把村庄用merge连通 
                             sum+=roads[j].P;
                             merge(roads[j].NUM_1,roads[j].NUM_2);
                       }
                       
                       for(k=1,S=0;k<=N;k++)//判断是否全部的村庄已经连通 
                       if(set[k]==k)
                       S+=1;
                       
                       if(S==1)//如果全部的村庄已连通，就跳出 
                       break;
                }
                printf("%d\n",sum);
            }
    }
    return 0;
}