问题描述
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 其中 1和2 2和4 4和5 3和7 7和8 8和9 有联系,怎么插入并查集呢?
用并查集的思想,我们会把2设为1的父亲,4设为2的父亲,5设为4的父亲,7设为3的父亲,8设为7的父亲,9设为8的父亲
这时并查集中就会出现三个组别,分别是5 <- 4 <- 2 <- 1 和 9 <- 8 <- 7 <- 3 和 6
我们可知,2的父亲的父亲为5,其它也同理,但6的父亲为6,其中5的父亲虽然没有指向,但并查集的规则是顶端的元素和单元素的父亲都是自己,这用代码怎么实现呢?首先初始化代码如图所示
我们先定义一个数组father[1000] = {0};
int Init(int n)
{
int i;
for (i = 0; i < n; i++)
father[i] = i;//并查集特点,顶端元素的父亲都是自己
}
查找代码:
int Find(int n)
{
if (father[x] == x)//一直找到该组中的顶端,返回顶端的值
return x;
else
return find(father[x]);//父亲的父亲的父亲。。。
}
合并代码:
int Union(int a, int b)
{
father[find(a)] = find(b);//找到a和b所在组中的顶端父节点,把其中一个父亲设为另一个父亲的父亲
}
前面的并查集有点垃圾,我们得优化一下它,怎么优化呢?在这里我们使用路径压缩的方法来优化,我们的并查集现在是这样的,1<-2<-3<-4,一串下去,要从4找到1就得一个一个遍历,很耗时间,我们可以把1设为2 3 4的父亲,怎么做呢,可以这样,father[2] = 1, father[3] = 1, father[4] = 1,当然我们不可能手动一个一个这样弄,看下面程序,我们只需变化一下Find函数
int Find(int x)
{
if (father[x] == x)
return x;
else
{
father[x] = Find(father[x]); //这样就可以找到顶端元素后,递归回退,把前面每个元素的父亲都设为顶端元素,这是并查集就不是一串了
return father[x];
}
}
题目解答代码如下:
#include <stdio.h>
#include <string.h>
int father[1000*1000] = {0};
int Find(int x)//并查集的搜索
{
if (father[x] == x)
return x;
else
{
father[x] = Find(father[x]);
return father[x];
}
}
void Union(int a, int b)//并查集的合并
{
father[Find(a)] = Find(b);
}
void Init(int n)//并查集的初始化
{
int i;
for (i = 1; i <= n; i++)
father[i] = i;
}
int main()
{
int m, n, k, a, b;
int i, j;
int count = 0;
scanf("%d%d%d", &m, &n, &k);
Init(m * n);
for (i = 0; i < k; i++)
{
scanf("%d%d", &a, &b);
Union(a, b);//把所有的元素都加入并查集
}
for (i = 1; i <= m * n; i++)
if (Find(i) == i)//父节点个数就是组的个数,可以通过查找总共有多少父节点来确定有多少组
count ++;
printf("%d\n", count);
return 0;
}