hdoj.1863 畅通工程【并查集】 20141129

原创于 2014-11-29 14:47:08 发布 · 311 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

HDoj（杭电）专栏收录该内容

290 篇文章

订阅专栏

畅通工程

Time Limit: 1000/1000 MS (Java/Others) Memory Limit: 32768/32768 K (Java/Others)

Total Submission(s): 18044 Accepted Submission(s): 7649

Problem Description

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

Input

测试输入包含若干测试用例。每个测试用例的第1行给出评估的道路条数 N、村庄数目M ( < 100 )；随后的 N
行对应村庄间道路的成本，每行给出一对正整数，分别是两个村庄的编号，以及此两村庄间道路的成本（也是正整数）。为简单起见，村庄从1到M编号。当N为0时，全部输入结束，相应的结果不要输出。

Output

对每个测试用例，在1行里输出全省畅通需要的最低成本。若统计数据不足以保证畅通，则输出“?”。

Sample Input

Sample Output

  
   3
?
  
  

  
  
   #include<stdio.h>
#include<algorithm>
using namespace std;
#define max 1010
int parent1[max],parent2[max],treedeep[max],n,m;
struct cc{
    int s;
    int e;
    int dis;
};
bool cmp(const cc& a,const cc& b){
    return a.dis<b.dis;
}
cc cou[max];
void make(){
    int i;
    for(i=1;i<=m;++i){
        parent1[i]=i;
        parent2[i]=i;
        treedeep[i]=0;
    }
}
int find1(int x){
    if(parent1[x]!=x)
        parent1[x]=find1(parent1[x]);
    return parent1[x];
}
int find2(int x){
    if(parent2[x]!=x)
        parent2[x]=find2(parent2[x]);
    return parent2[x];
}
void cr(int a,int b){
    int c=find1(a);
    int d=find1(b);
    if(c!=d)
        parent1[c]=d;
}
void unions(int x,int y){
    x = find2(x);
    y = find2(y);
    if(x==y) return;
    if(treedeep[x]>treedeep[y])
        parent2[y]=x;
    else{
        parent2[x]=y;
        if(treedeep[x]==treedeep[y])
            treedeep[y]++;
    }
}
int main(){
    int num,sum,i,x,y;
    while(scanf("%d%d",&n,&m),n){
        make();
        for(i=1;i<=n;++i){
            scanf("%d%d%d",&cou[i].s,&cou[i].e,&cou[i].dis);
            cr(cou[i].s,cou[i].e);
        }
        sort(cou+1,cou+n+1,cmp);
        num=0;
        for(i=1;i<=m;++i)
            if(parent1[i]==i)
                num++;
        if(num>1)
            printf("?\n");
        else{
            sum=0;
            for(i=1;i<=n;++i){
                x=find2(cou[i].s);
                y=find2(cou[i].e);
                if(x==y) continue;
                unions(x,y);
                sum+=cou[i].dis;
            }
            printf("%d\n",sum);
        }
    }
    return 0;
}