2018福州大学研究生复试机试题：畅通工程（并查集）

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 2
1 3
4 3

3 3
1 2
1 3
2 3

5 2
1 2
3 5

999 0
0

Sample Output

1
0
2
998

解题思路

1. 算法笔记p328并查集，将录入的道路数据，即两个城市视为一个集合。

经验总结

1. 所需建设的最少道路数可通过记录合并次数或统计城市集合数（需减1）来获得。

代码实现（C）

#include <stdio.h>

#define MaxSize 1000

int father[MaxSize];

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

// 查找
int findFather(int x) {
    if (x == father[x])                 // 找到根结点
        return x;
    else {
        int r = findFather(father[x]);  // 递归寻找father[x]的根结点r
        father[x] = r;                  // 将根结点r赋值给father[x]
        return r;                       // 返回根结点r
    }
}

// 合并
void Union(int a, int b) {
    int faA = findFather(a);    // 查找a的根结点,记为faA
    int faB = findFather(b);    // 查找b的根结点,记为faB
    if (faA != faB)             // 如果不属于同一个集合
        father[faA] = faB;      // 合并它们
}

// 统计城市合并次数
int countUnion(int N) {
    int count = 0;
    int faA = findFather(1);        // 令faA为合并后的根结点
    for (int i = 2; i <= N; ++i) {
        int faB = findFather(i);
        if (faA != faB) {
            father[faB] = faA;      // 合并到faA
            count++;                // 记录合并次数,即所需建设的道路数
        }
    }
    return count;
}

int main() {
    int N, M, a, b;
    while (~scanf("%d%d", &N, &M) && N) {
        // 初始化
        init(N);
        // 录入数据,合并城市
        for (int i = 0; i < M; ++i) {
            scanf("%d%d", &a, &b);
            Union(a, b);
        }
        // 合并不同集合城市,并打印所需建设的道路数
        printf("%d\n", countUnion(N));
    }
    return 0;
}

通过记录是否为根节点的方式判断集合个数，所需建设道路为集合个数减1：

#include <stdbool.h>

bool isRoot[MaxSize];

int countSet(int N) {
    // 初始化
    for (int i = 1; i <= N; ++i)
        isRoot[i] = false;
    // 记录是否为根节点
    for (int i = 1; i <= N; ++i)
        isRoot[findFather(i)] = true;
    // 统计集合数目
    int ans = 0;
    for (int i = 1; i <= N; ++i)
        ans += isRoot[i];
    // 所需建设的道路为城市集合数减1
    return ans - 1;
}