暑假前专题题解---数据结构---I

最新推荐文章于 2024-08-01 09:10:06 发布

风声sp

最新推荐文章于 2024-08-01 09:10:06 发布

阅读量203

点赞数

CC 4.0 BY-SA版权

分类专栏：暑假专题数据结构

本文链接：https://blog.youkuaiyun.com/CCCCTong/article/details/81392616

数据结构同时被 2 个专栏收录

12 篇文章

订阅专栏

暑假专题

11 篇文章

订阅专栏

本文介绍了一个基于并查集的数据结构实现，通过一个sum数组记录每个集合中元素的数量，实现快速查找和合并操作。适用于解决一些需要统计相同元素数量的问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

简单的并查集的应用，使用一个sum数组记录当前的编号的物品有多少和他相同的，这个数组有用的部分只是并查集的最高的节点的sum值，没次通过当前的数找到它的祖先即可，合并的时候sum相加。

代码如下

#include <bits/stdc++.h>
using namespace std;
const int maxn = 1e5 + 10;
int pre[maxn],sum[maxn];
int n,m,x,y,z,Q;
void init(){
    for (int i=0; i<n; i++) {
        pre[i] = i;
        sum[i] = 1;
    }
}
int find(int x){
    if (x == pre[x]) return x;
    return pre[x] = find(pre[x]);
}
void join(int x,int y){
    int fx = find(x),fy = find(y);
    if (fx != fy){
        pre[fx] = fy;
        sum[fy] += sum[fx];
    }
}
int main(){
    cin >> n >> m;
    init();
    for (int i=0; i<m; i++) {
        cin >> x >> y;
        join(x, y);
    }
    cin >> Q;
    for (int i=0; i<Q; i++) {
        cin >> z;
        z = find(z);
        cout << sum[z] << endl;
    }
    return 0;
}