欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

合根植物【蓝桥杯】

程序员文章站 2022-05-31 10:45:24
...

题目描述

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