蓝桥杯_PREV-54_合根植物

问题描述

　　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

样例说明

　　其合根情况参考下图

解题思路：并查集

#include <iostream>
#include <vector>
using namespace std;
class point{
public:
	int x, y;
	point(int x, int y) :x(x), y(y) {}
};
int find_root(int *p, int i){
	int j = i, k;
	while (p[i] != -1)
		i = p[i];
	while (j != i)
	{
		k = j;
		j = p[j];
		p[k] = i;
	}
	return i;
}//find_root函数查找根
void join(int *p, int i, int max){
	int k = i;
	while (i != -1){
		k = i;
		i = p[i];
		p[k] = max;
	}
}//改变一组节点为max
int main()
{
	int m, n, k, sum = 0;
	cin >> m >> n >> k;
	vector<point> all;
	int *p = new int[m * n];
	for (int i = 0; i < m*n; i++)
		p[i] = -1;
	for (int i = 0; i < k; i++){
		int val1, val2;
		cin >> val1 >> val2;
		all.push_back(point(val1-1, val2-1));
	}
	for (int i = 0; i < k; i++){
		int k1 = find_root(p, all[i].x);
		int k2 = find_root(p, all[i].y);
		if (k1 > k2)
			join(p, all[i].y, k1);
		else if(k1 < k2)
			join(p, all[i].x, k2);
	}
	for (int i = 0; i < m *n; i++)
		find_root(p, i);
	for (int i = 0; i < m *n; i++)
		if (p[i] == -1)
			sum++;
	cout << sum << endl;
    return 0;
}