合根植物【蓝桥杯】
程序员文章站
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;
}