并查集入门--畅通工程（HDU1232）

题目链接： http://acm.hdu.edu.cn/showproblem.php?pid=1232

畅通工程

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

Problem Description

某省调查城镇交通状况，得到现有城镇道路统计表，表中列出了每条道路直接连通的城镇。省政府“畅通工程”的目标是使全省任何两个城镇间都可以实现交通（但不一定有直接的道路相连，只要互相间接通过道路可达即可）。问最少还需要建设多少条道路？

 

Input

测试输入包含若干测试用例。每个测试用例的第1行给出两个正整数，分别是城镇数目N ( < 1000 )和道路数目M；随后的M行对应M条道路，每行给出一对正整数，分别是该条道路直接连通的两个城镇的编号。为简单起见，城镇从1到N编号。
注意:两个城市之间可以有多条道路相通,也就是说
3 3
1 2
1 2
2 1
这种输入也是合法的
当N为0时，输入结束，该用例不被处理。

 

Output

对每个测试用例，在1行里输出最少还需要建设的道路数目。

 

Sample Input

4 21 34 33 31 21 32 35 21 23 5999 00

 

Sample Output

102998

Hint

Hint

Huge input, scanf is recommended.

思路：这是我并查集入门的第二题，将提供的M条路连接的两个城市unite，最后输出根的数目-1，就是xjb套模板==

代码如下（函数用法见窝的模板==）：

#include <cstdio>
#include <set>
using namespace std;

int N,M,A,B;
int fa[1005],r[1005];

int fi(int x){
    return fa[x]==x?x:fa[x]=fi(fa[x]);
}

void unite(int x,int y){
    int p1=fi(x),p2=fi(y);
    if(p1==p2) return;
    if(r[p1]<r[p2]){
        fa[p1]=p2;
    }
    else{
        fa[p2]=p1;
    }
}

int main(){
    set<int> s;
    while(~scanf("%d%d",&N,&M)){
        if(N==0) break;
        s.clear();
        for(int i=1;i<=N;i++){
            fa[i]=i;
            r[i]=0;
        }
        for(int i=0;i<M;i++){
            scanf("%d%d",&A,&B);
            unite(A,B);
        }
        for(int i=1;i<=N;i++){
            if(!s.count(fi(i))){
                s.insert(fi(i));
            }
        }
        printf("%d\n",s.size()-1);
    }
}