原题链接:
http://poj.org/problem?id=1988
题意:
给定N个方块,排成一行,将它们编号1到N。再给出P个操作:
①M i j表示将i所在的那一堆移到j所在那一堆的顶上。
②C i表示一个询问,询问i下面有多少个方块。
思路:
带权并查集,一堆中最顶上的方块作为父节点,用dis[X] 统计X到父亲节点的距离,num[fa[X]]表示这个集合的大小,两者相减即为答案。
标程:
#include<cstdio>
#include<cstring>
#include<cmath>
#include<string>
#include<set>
#include<map>
#include<vector>
#include<iostream>
#include<algorithm>
using namespace std;
#define ll long long
const int inf = 0x3f3f3f3f;
const int maxn = 300010;
const double pi = acos(-1.0);
int pre[maxn],num[maxn],dis[maxn];
int n;
void init()
{
for(int i=0; i<n; i++)
{
pre[i]=i;
num[i]=1;
dis[i]=0;
}
}
int Find(int x)
{
if(x!=pre[x])
{
int t=pre[x];
pre[x]=Find(pre[x]);
dis[x]+=dis[t];
}
return pre[x];
}
int main()
{
while(~scanf("%d",&n))
{
init();
char op;
while(n--)
{
scanf(" %c",&op);
int x,y;
if(op=='M')
{
scanf("%d %d",&x,&y);
int fx=Find(x);
int fy=Find(y);
if(fx!=fy)
{
pre[fy]=fx;
dis[fy]=num[fx];
num[fx]+=num[fy];
}
}
else
{
scanf("%d",&x);
int fx=Find(x);
printf("%d\n",num[fx]-dis[x]-1);
}
}
}
return 0;
}