【并查集】HDU-1232 畅通工程

注解

1、典型的并查集。注意初始化操作、合并操作、查找操作以及路径压缩操作。
2、最后要达到的目的是所有城市连成一个集合。根据输入，每两个输入节点，判断是否在一个集合，如果不在就合并。最后输出的集合个数减一就是答案。
3、此题还可以扩展，当前是只查找到了集合的个数，如果要查找每个集合内的元素个数，把isRoot[findFather(i)] = 1;改成isRoot[findFather(i)]++;即可。然后遍历isRoot[]数组即可。

代码

#include <string.h>
#include <stdio.h>

const int MAX = 1000;
int N;
int father[MAX];
int isRoot[MAX];

void init() {
    for(int i=0; i<=N; i++) {
        father[i] = i;
    }
}

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;
    }
}

int main(){
    int m;
    scanf("%d", &N);
    while(N){
        init();
        scanf("%d", &m);
        for(int i=0; i<m; i++){
            int a, b;
            scanf("%d %d", &a, &b);
            Union(a, b);
        }
        memset(isRoot, 0, sizeof(isRoot));
        for(int i=1; i<=N; i++){
            isRoot[findFather(i)] = 1;
        }
        int sum = 0;
        for(int i=1; i<=N; i++){
            sum += isRoot[i];
        }
        printf("%d\n", sum-1);
        scanf("%d", &N);
    }
    return 0;
}

结果