九度_题目1512:用两个栈实现队列

这个思路实在太难想了。。。。

/*
 *str为PUSH,通过s1.push()使元素入栈;
 *str为POP时,s2非空,输出栈顶元素,s2.pop()弹出栈顶元素;若s2弹出元素后为空且s1不空,将s1中所有元素push进s2;
 *str为POP时,s2为空,若s1不空,将s1中所有元素push进s2,最后输出栈顶元素并通过<span style="font-family: Arial, Helvetica, sans-serif;">s2.pop()弹出栈顶元素</span>;
 */
题目描述:用两个栈来实现一个队列,完成队列的Push和Pop操作。
队列中的元素为int类型。
输入:
每个输入文件包含一个测试样例。
对于每个测试样例,第一行输入一个n(1<=n<=100000),代表队列操作的个数。
接下来的n行,每行输入一个队列操作:
1. PUSH X 向队列中push一个整数x(x>=0)
2. POP 从队列中pop一个数。
输出:
对应每个测试案例,打印所有pop操作中从队列pop中的数字。如果执行pop操作时,队列为空,则打印-1。
样例输入:
3
PUSH 10
POP
POP
样例输出:
10
-1
#include<iostream>
#include<string>
#include<stack>
using namespace std;
/*
 *str为PUSH,s1.push();
 *str为POP时,s2非空,输出栈顶元素s2.pop();若s2弹出元素后为空,若s1不空,将s1中所有元素push进s2;
 *str为POP时,s2为空,若s1不空,将s1中所有元素push进s2,最后输出栈顶元素;
 */
int main()
{
    int n;
    while(cin>>n)
    {
            stack<int>s1,s2;
            int num;
            string str;
        for(int i=0;i<n;i++)
        {
            cin>>str;
            if(str=="PUSH")
        {
            cin>>num;
            s1.push(num);
        }else if(str=="POP")
        {
            if(!s2.empty())
            {
                cout<<s2.top()<<endl;
                s2.pop();
                if(s2.empty())
                {
                    if(!s1.empty())
                    {
                        while(!s1.empty())
                        {
                            s2.push(s1.top());
                            s1.pop();
                        }
                    }
                     
                }
            }else
            {
                if(s1.empty())
                {cout<<-1<<endl;continue;}
                    else
                    {
                        while(!s1.empty())
                        {
                            s2.push(s1.top());
                            s1.pop();
                        }
                    }
                cout<<s2.top()<<endl;
                s2.pop();
 
            }
        }
        }
    }
}
/**************************************************************
    Problem: 1512
    User: hndxztf
    Language: C++
    Result: Accepted
    Time:580 ms
    Memory:1656 kb
****************************************************************/

你可能感兴趣的:(九度_题目1512:用两个栈实现队列)