-
题目描述:
-
用两个栈来实现一个队列,完成队列的Push和Pop操作。
队列中的元素为int类型。
-
输入:
-
每个输入文件包含一个测试样例。
对于每个测试样例,第一行输入一个n(1<=n<=100000),代表队列操作的个数。
接下来的n行,每行输入一个队列操作:
1. PUSH X 向队列中push一个整数x(x>=0)
2. POP 从队列中pop一个数。
-
输出:
-
对应每个测试案例,打印所有pop操作中从队列pop中的数字。如果执行pop操作时,队列为空,则打印-1。
思路:
首先声明两个栈,stack1:用于入队;stack2:用于出队。
当遇到PUSH操作(入队)时,将X压进栈stack1;
当遇到POP(出队)操作时,对栈stack2进行判断:
若stack2中有值,则打印stack2中栈顶的值并出栈;
若stack2中没值但stack1中有值,则将stack1中的所有值出栈并将这些值入栈stack2,而后打印stack2中栈顶的值并出栈;
若stack2中没值且stack1中也没值,则打印-1;
代码:
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
#define Max 100005
int main()
{
int n;
int stack1[Max],stack2[Max];
char opr[5];
int tmp;
while(scanf("%d",&n) != EOF)
{
int top1 = -1 , top2 = -1;
for(int i = 0 ; i < n ; i++)
{
scanf("%s",opr);
if(strcmp(opr,"PUSH") == 0)
{
scanf("%d",&tmp);
top1++;
stack1[top1] = tmp;
}
else if(strcmp(opr,"POP") == 0)
{
if(top2 > -1)
{
printf("%d\n",stack2[top2]);
top2--;
}
else if(top2 == -1 && top1 > -1)
{
while(top1 > -1)
{
top2++;
stack2[top2] = stack1[top1];
top1--;
}
printf("%d\n",stack2[top2]);
top2--;
}
else
{
printf("-1\n");
}
}
}
}
}