poj 1988 并查集(终于看懂一个了/(ㄒoㄒ)/~~)

时间:2022-01-23 08:15:30

题意:有几个stack,初始里面有一个cube。支持两种操作:1.move x y: 将x所在的stack移动到y所在stack的顶部。2.count x:数在x所在stack中,在x之下的cube的个数。

链接:点我

 /*
POJ 1988
*/
#include <stdio.h>
#include <iostream>
#include <algorithm>
#include <string.h>
using namespace std;
const int MAXN=;
int F[MAXN];
int num[MAXN];//堆的数目
int val[MAXN];//下面的个数
int find(int x)
{
if(F[x]==-)return x;
int tmp=find(F[x]);
val[x]+=val[F[x]];
return F[x]=tmp;
}
void bing(int u,int v)//将u所在的堆放在v上面,注意顺序
{
int t1=find(u),t2=find(v);
if(t1!=t2)
{
F[t1]=t2;
val[t1]=num[t2];
num[t2]+=num[t1];
}
}
int main()
{
int P;
int u,v;
char op[];
while(scanf("%d",&P)==)
{
for(int i=;i<MAXN;i++)
{
F[i]=-;
val[i]=;
num[i]=;
}
while(P--)
{
scanf("%s",&op);
if(op[]=='C')
{
scanf("%d",&u);
find(u);
printf("%d\n",val[u]);
}
else
{
scanf("%d%d",&u,&v);
bing(u,v);
}
}
}
return ;
}