0
点赞
收藏
分享

微信扫一扫

蓝桥杯之并查集的应用

登高且赋 2022-03-30 阅读 81

题目描述

w 星球的一个种植园,被分成m×n 个小格子(东西方向 m 行,南北方向 n 列)。每个格子里种了一株合根植物。

这种植物有个特点,它的根可能会沿着南北或东西方向伸展,从而与另一个格子的植物合成为一体。

如果我们告诉你哪些小格子间出现了连根现象,你能说出这个园中一共有多少株合根植物吗?

输入描述

第一行,两个整数 m,n,m,n,用空格分开,表示格子的行数、列数(1≤m,n≤1000)。

接下来一行,一个整数 k (0≤k≤10^5 ),表示下面还有 k 行数据。

接下来 kk 行,每行两个整数 a,b,表示编号为 a 的小格子和编号为 bb 的小格子合根了。

格子的编号一行一行,从上到下,从左到右编号。

比如: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

其合根情况参考下图:

运行限制

  • 最大运行时间:2s
  • 最大运行内存: 256M

代码:

//合根植物
#include<iostream>
#include<iomanip>
#include<algorithm>
using namespace std;

int m, n;
int pre[100005] = { 0 };
int temp[100005] = { 0 };
void init(int n)
{
    for (int i = 1; i <= n; i++)
    {
        pre[i] = i;//初始化
    }
}
int find(int x)//寻找根结点
{
    if (pre[x] == x) return x;
    return pre[x] = find(pre[x]);
}
void unit(int x, int y)
{
    int rx = find(x);
    int ry = find(y);
    if (rx != ry) pre[rx] = ry;
}

int main()
{
    int a, b, k;
    cin >> m >> n >> k;
    init(m * n);//初始化m*n的易威数组
    for (int i = 0; i < k; i++)
    {
        cin >> a >> b;
        unit(a, b);//联合
    }
    int cnt = 0;
  
    for (int i = 1; i <= m * n; i++)
    {
        temp[find(i)] = 1;//寻找父亲,并且置为1
    }
    for (int i = 1; i <= m * n; i++)
    {
        if (temp[i] == 1) cnt++;//计算父亲的数量
    }
    cout << cnt << endl;
    
}
举报

相关推荐

0 条评论