算法笔记-问题 B: 畅通工程

问题 B: 畅通工程

题目描述

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

输入

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

输出

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

样例输入 Copy

5 3
1 2
3 2
4 5
0

样例输出 Copy

1

代码：

#include <bits/stdc++.h>

using namespace std;

const int maxn = 1000;

int father[maxn];
bool isRoot[maxn];
int n, m;

/*寻找祖先*/
int findFather(int x){
    int a = x;
    while(x!=father[x]){
        x = father[x];
    }
    while(a!=father[a]){
        int z = a;
        a = father[a];
        father[z] = x;
    }
    return x;
}

/*合并*/
void Union(int a, int b){
    int faA = findFather(a);
    int faB = findFather(b);
    if(faA != faB){
        father[faA] = faB;
    }
}

/*初始化*/
void init(int n){
    for(int i=1; i<=n; i++){
        father[i] = i;
        isRoot[i] = false;
    }
}

int main(){
    while(scanf("%d", &n)!=EOF){
        if(n==0) break;
        scanf("%d", &m);
        init(n);
        for(int i=0; i<m; i++){
            int a, b;
            scanf("%d%d", &a, &b);
            Union(a, b);
        }
        for(int i=1; i<=n; i++){
            isRoot[findFather(i)] = true;
        }
        int ans = 0;
        for(int i=1; i<=n; i++){
            ans += isRoot[i];
        }
        printf("%d\n", ans-1);
    }
	return 0;
}