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

简单的并查集的应用,使用一个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;
}

猜你喜欢

转载自blog.csdn.net/CCCCTong/article/details/81392616