Description
对两个栈A和B有以下三种操作:
push A x:x入A栈
pop A:A栈顶元素出栈
merge A B:合并A和B,合并后B栈为空栈,合并后的栈按插入时间放置元素,插入时间最后的在栈顶
Input
多组用例,每组用例首先输入操作次数n,之后n行每行一个操作,保证不会对空栈进行pop操作,以n=0结束输入(n<=10^5)
Output
对于每次pop操作,输出出栈元素
Sample Input
4
push A 1
push A 2
pop A
pop A
9
push A 0
push A 1
push B 3
pop A
push A 2
merge A B
pop A
pop A
pop A
9
push A 0
push A 1
push B 3
pop A
push A 2
merge B A
pop B
pop B
pop B
0
Sample Output
Case #1:
2
1
Case #2:
1
2
3
0
Case #3:
1
2
3
0
Solution
push和pop操作普通的栈即可实现,但是merge操作的加入使得只开两个栈无法在时限内解决这个问题,我们可以再开一个栈C保存两个栈merge后的结果,这样每次merge的时候就是将A栈和B栈中的元素放在C栈后面然后将A栈和B栈清空,每次pop A的时候如果A栈非空那么直接输出A栈栈顶元素,如果A栈是空的说明之前进行过merge A B操作(因为一定不会对空栈进行操作),那么C栈的栈顶元素其实就是实际A栈的栈顶元素,pop B同理,这样时间复杂度就是O(n)
Code
#include<cstdio>
#include<iostream>
#include<cstring>
#include<algorithm>
using namespace std;
#define maxn 111111
struct node
{
int v,t;
node(){};
node(int _v,int _t)
{
v=_v,t=_t;
}
}a[maxn],b[maxn],c[maxn];
int n,x,pa,pb,pc,Case=1;
char s1[11],s2[11],s3[11];
int main()
{
while(~scanf("%d",&n),n)
{
printf("Case #%d:\n",Case++);
pa=pb=pc=0;
for(int i=1;i<=n;i++)
{
scanf("%s%s",s1,s2);
if(s1[1]=='u')
{
scanf("%d",&x);
if(s2[0]=='A')a[pa++]=node(x,i);
else b[pb++]=node(x,i);
}
else if(s1[0]=='m')
{
scanf("%s",s3);
int aa=0,bb=0;
while(aa<pa&&bb<pb)
{
if(a[aa].t<b[bb].t)c[pc++]=a[aa++];
else c[pc++]=b[bb++];
}
while(aa<pa)c[pc++]=a[aa++];
while(bb<pb)c[pc++]=b[bb++];
pa=pb=0;
}
else
{
if(s2[0]=='A')
{
if(pa)printf("%d\n",a[--pa].v);
else printf("%d\n",c[--pc].v);
}
else
{
if(pb)printf("%d\n",b[--pb].v);
else printf("%d\n",c[--pc].v);
}
}
}
}
return 0;
}