【蓝桥杯】合根植物(并查集)

问题描述
　　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
样例输入
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

思路分析：

其实就是并查集。每个合根植物都是一个连通分量，并查集求连通分量个数。(注意6也算一个合根植物)

#include <iostream>
#define MAX 1000010
using namespace std;

int pre[MAX];
int h[MAX];

void init()
{
	for(int i = 0;i < MAX;i++)
	pre[i] = i, h[i] = 1;
}

int Find(int x)
{
	if(pre[x] == x) return x;
	return pre[x] = Find(pre[x]);
}

void Union(int x, int y)
{
	int rx = Find(x);
	int ry = Find(y);
	if(rx == ry) return;
	if(h[rx] > h[ry]) pre[ry] = rx;
	else if(h[ry] > h[rx]) pre[rx] = ry;
	else pre[ry] = rx, h[rx]++; 
}

int main()
{
	int m, n, k;
	while(cin>>m>>n>>k)
	{
		int cnt = 0;
		init();
		for(int i = 0;i < k;i++)
		{
			int x, y;
			cin>>x>>y;
			Union(x,y);
		}
		for(int i = 1;i <= m*n;i++)
		{
			if(pre[i] == i)
			cnt++;
		}
		cout<<cnt<<endl;
	}
	return 0;
}