HDOJ 1232 畅通工程

来源：互联网发布：mac桌面时钟日历插件编辑：程序博客网时间：2024/05/01 20:17

超级传送门：

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