关闭

HDU 5818 Joint Stacks(栈)

102人阅读 评论(0) 收藏 举报
分类:

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;
}
0
0

查看评论
* 以上用户言论只代表其个人观点,不代表CSDN网站的观点或立场
    个人资料
    • 访问:393045次
    • 积分:20208
    • 等级:
    • 排名:第426名
    • 原创:1692篇
    • 转载:0篇
    • 译文:0篇
    • 评论:64条
    最新评论