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