HDOJ 1232 畅通工程

超级传送门：

http://acm.hdu.edu.cn/showproblem.php?pid=1232

赤裸裸的并查集，初始时有n个连通分量，每输入一组路线就进行一次并查集合并操作，最后统计剩余连通分量的个数count，所需的最少道路数为count-1。

#include<stdio.h>
int find(int x);
void merge(int a,int b);
int set[1010];
int main () {
    //freopen("1.txt","r",stdin);
    int n,m,i,j,a,b,count,ok,br[1010];
    while (scanf("%d%d",&n,&m) != EOF && n) {
        count = 0;
        for (i=1;i<=n;i++) set[i] = i;//初始化并查集
        for (i=1;i<=m;i++) {
            scanf("%d%d",&a,&b);
            merge(a,b);
        }
        for (i=1;i<=n;i++) {
            ok = 1;
            for (j=0;j<count;j++) {
                if (find(i)==br[j]) {
                    ok = 0;
                    break;
                }
            }
            if (ok) br[count++] = find(i);//增加连通分量
        }
        printf("%d\n",count-1);//还需要修建的路：连通分量数-1
    }
    return 0;
}
int find(int x) {
    while (set[x]!=x) x = set[x];
    return x;
}
void merge(int a,int b) {
    if (find(a)<find(b)) set[find(b)] = a;
    else if (find(a)>find(b)) set[find(a)] = b;
}

下面采用路径压缩和树深比较来优化并查集。

优化并查集后的代码：

#include<stdio.h>
int find(int x);
void merge(int a,int b);
int set[1010],height[1010];
int main () {
    //freopen("1.txt","r",stdin);
    int n,m,i,j,a,b,count,ok,br[1010];
    while (scanf("%d%d",&n,&m) != EOF && n) {
        count = 0;
        for (i=1;i<=n;i++) {
            set[i] = i;//初始化并查集
            height[i] = 1;//初始化树深
        }
        for (i=1;i<=m;i++) {
            scanf("%d%d",&a,&b);
            merge(a,b);
        }
        for (i=1;i<=n;i++) {
            ok = 1;
            for (j=0;j<count;j++) {
                if (find(i)==br[j]) {
                    ok = 0;
                    break;
                }
            }
            if (ok) br[count++] = find(i);//增加连通分量
        }
        printf("%d\n",count-1);//还需要修建的路：连通分量数-1
    }
    return 0;
}
int find(int x) {
    int r = x,i = x,j;
    while (set[r] != r) r = set[r];
    while (i != r) {
        j = set[i];
        set[i] = r;
        i = j;
    }
    return r;
}
void merge(int a,int b) {
    int fa = find(a),fb = find(b);
    if (height[a]==height[b]) {
        height[a] += 1;
        set[fa] = fb;
    } else if (height[a]<height[b]) {
        set[fa] = fb;
    } else {
        set[fb] = fa;
    }
}