这个思路实在太难想了。。。。
/*
*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
****************************************************************/