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