您的位置:首页 > 其它

Poj 1988 Cube Stacking

2016-06-30 10:18 281 查看
题意:有n个元素,开始每个元素自己一栈,有两种操作,将含有元素x的栈放在含有y的栈的顶端,合并为一个栈。第二种操作是询问含有x元素下面有多少个元素。

解析:直接上代码

#include<cstdio>
#include<cstring>
#include<string>
#include<algorithm>
using namespace std;
const int maxn=100009;
int set[maxn];
int top[maxn];
int cnt[maxn];
int n;
int find(int x)
{
if(set[x]<0)
return x;
if(set[set[x]]>=0)
{
int fa=set[x];
set[x]=find(fa);
cnt[x]+=cnt[fa];
}
return set[x];
}
void merge(int x,int y)
{
x=find(x);
y=find(y);
set[x]=y;
find(top[y]);
cnt[x]=cnt[top[y]]+1;
top[y]=top[x];
}
int main()
{
while(scanf("%d",&n)!=EOF)
{
memset(set,-1,sizeof(set));
memset(cnt,0,sizeof(cnt));
for(int i=0;i<maxn;i++)
top[i]=i;
char st[23];
for(int i=0;i<n;i++)
{
scanf("%s",st);
if(st[0]=='M')
{
int x,y;
scanf("%d%d",&x,&y);
merge(x,y);
}
else
{
int x;
scanf("%d",&x);
find(x);
printf("%d\n",cnt[x]);
}
}
}
return 0;

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签:  poj