HDU－1233畅通工程(并查集&最小生成树)

HDU－1233畅通工程(并查集&最小生成树)

主要用的是并查集

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
做题思路：
起初让每个村庄自身各为一个集合，设根结点就是自己本身（设根结点的目的就是为了方便查询并合并），同时又用到了最小生成树的克鲁斯卡尔算法，即让每两个村庄的距离从小到大依次排序，然后从小开始遍历两个村庄，先判断这两个村庄的根结点是否相同； 如果不同，说明在不同的集合，则开始合并他们，并把他们的距离加起来； 如果根结点相同，则在同一个集合，这时候就不能连他们了（克鲁斯卡尔算法中不能产生回路）。

#include <iostream>
#include <cstring>
#include <algorithm>
using namespace std;
const int maxn=103;
int f[maxn];
struct ss{
 int x,y,z;
}a[5000]; 
/* run this program using the console pauser or add your own getch, system("pause") or input loop */
bool cmp(ss c,ss d){
 return c.z<d.z;
}
int find(int x){
 if(f[x]!=x){
  f[x]=find(f[x]);
 }
 return f[x];
}
int main(int argc, char** argv){
 int n,s;
 while(cin>>n&&n){
  for(int i=1;i<=n;i++){
   f[i]=i;
  }
  int N=n*(n-1)/2;
  for(int i=1;i<=N;i++){
   cin>>a[i].x>>a[i].y>>a[i].z;
  }
  sort(a+1,a+1+N,cmp);
  int s=0;
  for(int i=1;i<=N;i++){
   int b=find(a[i].x);
   int c=find(a[i].y);
   if(b==c) continue;
   f[c]=b;
   s+=a[i].z;
  }
  printf("%d\n",s);
 }
 return 0;
}