合根植物

题目描述

题目来源：2017蓝桥杯国赛

w 星球的一个种植园，被分成 m×n个小格子（东西方向 m 行，南北方向 n 列）。每个格子里种了一株合根植物。

这种植物有个特点，它的根可能会沿着南北或东西方向伸展，从而与另一个格子的植物合成为一体。

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

输入描述

第一行，两个整数 m,n，用空格分开，表示格子的行数、列数（1≤m,n≤1000）。

接下来一行，一个整数 k (0≤k≤10^5 )，表示下面还有 k 行数据。

接下来 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

 

输出描述

输出植物数量。

输入输出样例

示例

输入

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

 

输出

5

 

样例说明

其合根情况参考下图：

运行限制

• 最大运行时间：2s
• 最大运行内存: 256M

思路：

1. 使用并查集解决问题；用一个数组ab（每个元素都初始化为其下标，即最初每个数指向自己，或者说父亲结点为自己）来存放输入的两个整数a、b之间的关系；如ab[x]=y;表示x指向y，或者说x的父亲结点为y；

2. 对于每一组输入a、b；通过find()函数分别寻找其指向的结点（即父亲结点）fa和fb，若a，b有共同的父亲，即fa=fb，则不作操作，否则将a、b和为一体，可以把fa指向fb（即fb为fa的父亲结点），也可以把fb指向fa（即fa为fb的父亲结点）;

3. 找出所有结点的的根结点（即父亲结点的父亲结点……直到没有父亲结点的那个结点），然后统计根结点的数目即为最终答案（每个根结点和直接或者间接指向自己的结点算是一株植物）；

 

//2017g 合根植物     vxgzh：xtsn
#include <iostream>
using namespace std;
#include <set>
#define N 1005

int ab[N*N]={0};   //存放的是父节点 

int find(int a)
{
	if(ab[a]==a)   //a指向它自己 ，即a的根是自己 
		return a;
	else
		return ab[a]=find(ab[a]);    //a的根是find(ab[a]),并作修改 
}
int main()
{ 
	int n,m,k,a,b;	
	cin>>n>>m; 
	for(int i=1;i<=n*m;i++)   //父节点的初始化 
		ab[i]=i;		
	cin>>k;
	for(int i=0;i<k;i++)
	{
		cin>>a>>b;
		int fa = find(a);   //找到a的根 fa 
		int fb = find(b);   //找到b的根 fb
		if(fa!=fb)    //没有共同的根 
	            ab[fa]=fb;      //  ab[fb]=fa;也可以 
	//将a的根fa指向b的根fb，即把a的根fa的根（原来fa的根为它自己）修改fb；		
	}
	set <int> se;
	for(int i=1;i<=n*m;i++)
		se.insert(find(i));   //找到每个结点的根； 

	cout<<se.size()<<endl;   //不同根的数量 
		
	
	return 0;
}