题目描述:
用两个栈来实现一个队列,完成队列的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
解题:
用两个栈实现一个队列,队列的性质是先进先出。栈是后进先出,所以对于两个栈,“倒一下”就可以了。
s1,s2为两个栈,
当PUSH的时候,s1总是进栈;
当POP的时候,s2如果为空(不空不能倒,会影响顺序),将s1中元素全倒入s2;
s2中有元素,输出即可,否则-1
#include <iostream>
#include <cstdio>
#include <stack>
using namespace std;
stack<int> s1,s2;
char cmd[10];
int main()
{
int n;
while(scanf("%d",&n) != EOF)
{
while(n--)
{
scanf("%s", cmd);
if(cmd[1] == 'U'){ //push
int val; scanf("%d", &val);
s1.push(val);
}
else if(cmd[1] == 'O'){ //pop
if(s2.empty()){
while(!s1.empty()) { int t = s1.top(); s1.pop(); s2.push(t); }
}
if(!s2.empty()) { printf("%d\n",s2.top()); s2.pop();}
else printf("-1\n");
}
}
while(!s1.empty()) s1.pop();
while(!s2.empty()) s2.pop();
}
return 0;
}