并查集的题目
【题目大意】
有n个元素,开始每个元素自己 一栈,有两种操作,将含有元素x的栈放在含有y的栈的顶端,合并为一个栈。第二种操作是询问含有x元素下面有多少个元素。
用sum数组储存每个栈中的元素个数,每次合并的时候将sum加到 父亲节点,也就是每个栈的最底部。
用under数组储存当前节点下面有多少元素。每次合并的时候,就可以将顶端元素的under赋值为父节点也就是栈最底部的sum。
void Union(int x,int y){
int xr = find(x);
int yr = find(y);
if(xr==yr) return;
father[xr]=yr;
under[xr]=sum[yr];
sum[yr]+=sum[xr];
}
在查询的时候,运用递归的思想,从底部往上加under。
int find(int x){
if(x==father[x])
return father[x];
int tmp = find(father[x]);
under[x]+=under[father[x]]; //仔细想想
father[x]=tmp;
return tmp;
}
【源代码】
#include <iostream> //父亲节点在栈的底部
#include <cstdio>
using namespace std;
const int maxn = 30000+10;
int father[maxn];
int under[maxn];
int sum[maxn];
void init(){
for(int i=0;i<maxn;i++){
father[i]=i;
under[i]=0;
sum[i]=1;
}
}
int find(int x){
if(x==father[x])
return father[x];
int tmp = find(father[x]);
under[x]+=under[father[x]];
father[x]=tmp;
return tmp;
}
void Union(int x,int y){
int xr = find(x);
int yr = find(y);
if(xr==yr) return;
father[xr]=yr;
under[xr]=sum[yr];
sum[yr]+=sum[xr];
}
int main(){
int n;
while(scanf("%d",&n)!=EOF){
char ch;
int a,b;
init();
for(int i=0;i<n;i++){
scanf(" %c",&ch);
if(ch=='M'){
scanf("%d%d",&a,&b);
Union(a,b);
}
else{
scanf("%d",&a);
int x=find(a); //只是用来累加一下under
printf("%d\n",under[a]);
}
}
}
return 0;
}