HDU 1232 畅通工程 笔记

畅通工程

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

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

Hint
Hint

Huge input, scanf is recommended.

Source
浙大计算机研究生复试上机考试-2005年

Recommend
JGShining | We have carefully selected several similar problems for you: 1233 1272 1875 1879 1213

---

说实话，以前虽然学过并查集，但是对并查集的应用知之甚少。畅通工程这个题算是最经典的也是最基础的并查集习题。

题目要求”最少还需要建设多少条道路？ “，在并查集中，只要求出连通分量个数s，就可以知道还需要建设多少条路，因为只要把全部连通分量连在一起，就能使全部城市连通，所以还需的道路条数为s-1

虽然推荐用scanf，但cin也可以ac

#include <iostream>
using namespace std;
int pre[10000];

inline void init(int n);
void init(int n){
    for (int i = 1; i <= n; ++i)
        pre[i] = i;
}
inline int find(int n);
int find(int n){
    while (pre[n] != n)
        n = pre[n];
    return n;
}
inline void combine(int a, int b);
void combine(int a, int b){
    int t_a = find(a);
    int t_b = find(b);
    if (t_a != t_b)
        pre[t_a] = t_b;
}
int ans(int n){
    int r = 0;
    for (int i = 1; i <= n; ++i)
        if (pre[i] == i)
            ++r;
    return r;
}
int main(){
    int n, m;
    while (cin >> n){
        if (n == 0)
            break;
        cin >> m;
        init(n);
        for (int i = 0; i < m; ++i){
            int a, b;
            cin >> a >> b;
            combine(a, b);
        }
        cout << ans(n) - 1 << endl;
    }
    return 0;
}

下面是路径压缩优化过的算法：

#include <iostream>
using namespace std;
int pre[10000];
int ran[10000]; //值越大说明高度越高（从底向上）

inline void init(int n);
void init(int n){
    for (int i = 1; i <= n; ++i){
        pre[i] = i;
        ran[i] = 1;
    }
}
int find(int x){
    if (pre[x] == x)
        return x;
    return pre[x] = find(pre[x]);   //非常巧妙的做法
}
inline void combine(int a, int b);
void combine(int a, int b){
    int t_a = find(a);
    int t_b = find(b);
    if (t_a == t_b)
        return;
    if (ran[t_a] > ran[t_b])
        pre[t_b] = t_a;
    else {
        if (ran[t_a] == ran[t_b])
            ++ran[t_b];
        pre[t_a] = t_b;
    }
}
int ans(int n){
    int r = 0;
    for (int i = 1; i <= n; ++i)
        if (pre[i] == i)
            ++r;
    return r;
}
int main(){
    int n, m;
    while (cin >> n){
        if (n == 0)
            break;
        cin >> m;
        init(n);
        for (int i = 0; i < m; ++i){
            int a, b;
            cin >> a >> b;
            combine(a, b);
        }
        cout << ans(n) - 1 << endl;
    }
    return 0;
}

最后再补充一句：
关于路径压缩的做法，有递归和非递归的形式，某些问题中，一旦递归深度太大，会导致stack overflow，所以有时候用非递归的形式可能更合适：

int find(int x){
    int r, k, j;
    r = x;
    while (r != pre[r])
        r = pre[r];
    k = x;
    while (k != r){
        j = pre[k];
        pre[k] = r;
        k = j;
    }
    return r;
}

参考资料：

• http://blog.csdn.net/lttree/article/details/23820679
• https://www.cnblogs.com/xzxl/p/7226557.html