合根植物【蓝桥杯】

题目描述

问题描述
　　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

考点：并查集
遇到问题：
1.数组开小了
2.因为会出现祖先两边倒的情况，最后需要再遍历一遍数组调用findfa函数统一一下（之前还给每对a，b排了序，然后运行一直出问题）
3.用scanf和printf代替cin和cout可以提高运行速度。

#include<iostream>
#include<vector>
#include<cstdio>
#include<map>
#include<set>
#include<algorithm>
using namespace std;
int m,n,k,cnt=0,ans=0;
int v[1000005],vis[1000005];
int findfa(int k)
{
	if(k == v[k])
		return k;
	return v[k] = findfa(v[k]);
}
int main()
{
	cin >> m >> n >> k;
	for(int i=1;i<=m*n;i++)
		v[i] = i;
	while(k--)
	{
		int p,q;
		scanf("%d%d",&p,&q);
//		cin >> p >> q;
		int a = findfa(p);
		int b = findfa(q);
		if(a!=b) v[b] =a;
	}
	set<int> s; //其实可以直接遍历数组，即可提高运行速度也可减少空间利用
	for(int i=1;i<=n*m;i++)
		s.insert(findfa(i));
	printf("%d",s.size());
//	cout << s.size();
	return 0;
} 

附上一份大佬的比较标准的写法：

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e6+100;
int pre[maxn];
void initUFset(int n) //Union_Find_Set 并查集 
{
	for(int i=0;i<=n;++i) pre[i] = i;
}
int Find(int x)
{
	return x==pre[x]?x:pre[x]=Find(pre[x]);
}
void join(int x,int y)
{
	int fx = Find(x);
	int fy = Find(y);
	if(fx!=fy) pre[fy] = fx;
}
int main()
{
	int m,n,k,x,y,tot;
	scanf("%d%d%d",&m,&n,&k);
	tot = m*n;
	initUFset(tot);
	while(k--)
	{
		scanf("%d%d",&x,&y);
		join(x,y);
	} 
	int ans = 0;
	for(int i=1;i<=tot;++i)
		if(Find(i)==i) 
			++ans;
	printf("%d\n",ans);	
	return 0;	
}