并查集算法学习

蓝桥杯题目:合根植物

问题描述

w星球的一个种植园,被分成 m * n 个小格子(东西方向m行,南北方向n列)。每个格子里种了一株合根植物。
这种植物有个特点,它的根可能会沿着南北或东西方向伸展,从而与另一个格子的植物合成为一体。

如果我们告诉你哪些小格子间出现了连根现象,你能说出这个园中一共有多少株合根植物吗?

输出格式

第一行,两个整数m,n,用空格分开,表示格子的行数、列数(1<m,n<1000)。
接下来一行,一个整数k,表示下面还有k行数据(0<k<100000)
接下来k行,第行两个整数a,b,表示编号为a的小格子和编号为b的小格子合根了。

格子的编号一行一行,从上到下,从左到右编号。
比如:5 * 4 的小格子,编号:
1 2 3 4
5 6 7 8
9 10 11 12
13 14 15 16
17 18 19 20

样例输入

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
5 4
16
2 3
1 5
5 9
4 8
7 8
9 10
10 11
11 12
10 14
12 16
14 18
17 18
15 19
19 20
9 13
13 17

样例输出

1
5

其和根情况如下图

一直把6这个点给忽略了,6也可以算一个根。

解决方案

这里采用并查集算法。

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
#include <stdio.h>
#include <string.h>
int* all;
int count = 0 ;

void union1(int a, int b){
all[b]=a;
}

int find(int a){
if(all[a]==0) return a;
all[a] = find(all[a]);
return all[a];

}
void func(int a , int b){

int ra = find(a);
int rb = find(b);
if(ra==rb) return ;
union1(ra,rb);
count ++;

}
int main(int argc, char *argv[]) {
int n = 0, m = 0,sum = 0, a = 0, b = 0;
scanf("%d%d",&m,&n);
scanf("%d",&sum);
all = new int[n*m + 1];
memset(all,0,(m*n+1)*sizeof(int));
for(int i = 0; i < sum ; i ++){
scanf("%d%d",&a,&b);
func(a,b);
}
printf("%d",m*n-count);
return 0;
}

就是把相同的两个点连接起来,最后分成多少个分组,这里使用的是并查集算法,百度有并查集算法的解释,这里只贴出这个例子