COCI2014 stogovi

题目描述:

Mirko在玩堆栈游戏。开始他有一个空的堆栈,编号为0.在第i步(1<=i<=300000),他会选择一个编号为v的堆栈,做如下操作:
1.a v 表示将v号堆栈复制一份,新栈的编号为i,并将元素i压入新栈的栈顶。
2. b v 表示将v号堆栈复制一份,新栈的编号为i,将新栈的栈顶元素弹出。
3.c v w 将v号堆栈复制一份,编号为i,并比较第v号和第w号堆栈中有多少相同的数。

输入:

第一行一个整数n,表示有n步。
接下来n步,每步表示一个操作,如上所述。

输出:

对所有的b操作和c操作,输出结果,每行一个。b操作需要输出该栈移除的元素,c操作表示两个堆栈的相同的数的个数。

样例输入:

5
a 0
a 1
b 2
c 2 3
b 4

样例输出:

2
1
2

分析:

这是一道比较神的题。用堆栈打掩护,其实是一道LCA的题。在考试的时候虽然隐约感觉这些操作与之前的堆栈有所联系,但没有明确的想出来怎样建图(以前没怎么独立做过LCA的题),乱码了200行,一分没有……下来自己又码了一边,居然一边就A了,可能跟我抄了接近十遍的NOIP2013 day2 truck 有很大的关系吧。“a”操作我们将它翻译为在相关节点连一条父子边,i为输入的x的儿子;“b”操作翻译为将i与输入的x的父亲合并在一起,并输出x的栈顶元素;“c”操作翻译为将i与输入的x1合并,寻找i与输入的x2的最近公共祖先,并输出它的深度。因为题目中给出信息每一次压入堆栈的元素并不相同,所以不用担心在lca(i,x2)后面会有相同的数,就可以直接输出deep[lca(i,x2)]。

code:

#include
#include
#include
#define MAXN 300000
#define MAXD 20
using namespace std;
int fa[MAXN+5][MAXD+5],deep[MAXN+5];
int unit[MAXN+5],top[MAXN+5];
int n,d1,d2;
char o[2];
int find(int x){return unit[x]==x?x:unit[x]=find(unit[x]);}
void dfs(int u,int f)
{
	int i;
	fa[u][0]=f;
	deep[u]=deep[f]+1;
	for(i=1;i<=MAXD;i++)
	{
		if(deep[u]<(1<=0;i--)
		if(fa[u][i]!=fa[v][i])
		{
			u=fa[u][i];
			v=fa[v][i];
		}
	if(u==v)
		return u;
	return fa[u][0];
}
int main()
{
	int i;
	scanf("%d",&n);
	memset(fa,-1,sizeof fa);
	for(i=1;i<=n;i++)
		unit[i]=i;
	for(i=1;i<=n;i++)
	{
		scanf("%s",o);
		if(o[0]=='a')
		{
			scanf("%d",&d1);
			d1=find(d1);
			dfs(i,d1);
			top[i]=i;
		}
		else if(o[0]=='b')
		{
			scanf("%d",&d1);
			d1=find(d1);
			int f=find(fa[d1][0]);
			top[i]=top[f];
			dfs(i,f);
			unit[i]=f;
			printf("%d\n",top[d1]);
		}
		else if(o[0]=='c')
		{
			scanf("%d%d",&d1,&d2);
			d1=find(d1),d2=find(d2);
			top[i]=top[d1];
			int f1=find(fa[d1][0]),f2=find(d2);
			dfs(i,f1);
			unit[i]=d1;
			int fi=find(i);
			printf("%d\n",deep[lca(fi,f2)]);
		}
	}
}


你可能感兴趣的:(COCI2014 stogovi)