如何用两个栈实现一个队列

最新推荐文章于 2025-03-10 19:28:37 发布

Appiloveyou

最新推荐文章于 2025-03-10 19:28:37 发布

阅读量488

点赞数

CC 4.0 BY-SA版权

文章标签：算法

本文链接：https://blog.youkuaiyun.com/Appiloveyou/article/details/47152987

这篇博客讨论了如何使用两个栈来模拟队列的操作。作者在解决这个问题时遇到了超时问题，最初的想法是将入队操作放入主栈A，出队时将A栈元素转移至辅助栈B，但在实际实现中发现这种方法效率低下。通过查看解析，作者理解到正确的思路是在出队时首先检查辅助栈B，如果为空则将主栈A的所有元素转移到B栈，若此时B栈仍为空则返回-1，否则从B栈顶部出队。作者最终实现了AC的解决方案。

今天在九度OJ 上做题，做到了这个，感觉神坑啊，我刚开始一直WA ，初步估计时gets的原因，感觉这个函数有时候过不去OJ ，不知道为啥，反正POJ ，HDOJ做的时候没碰到这个问题，然后就是scanf的用法，遇到空格，\t ，回车时候结束，当时搞了好久，有点汗颜。

最开始的想法就是两个栈，这里假定就是A栈和B栈，A栈是主栈，B栈是辅助栈，入队就是入A栈，出队就先把A元素出栈入B栈直到A栈只剩一个元素，然后出栈，接着把B栈元素全部赋值过来，思想很简单，但是超时。

后来看了解析，大致的思想是入队是入A栈，出队的时候先看B栈，如果B栈空，把A栈的所有元素出栈入B 栈，在进行检查，如果此时B 栈空，则输出-1，否则栈顶元素出栈。

AC了，感觉这个思想明显比上一个要好。

下面是代码

#include<stdio.h>
#include<iostream>
#include<stdlib.h>
#include<algorithm>
#include<string.h>
#define inf 1000000000
using namespace std;
int main()
{
    int a[100005],b[100005];
    int top1=-1;
    int top2=-1;
    char str[30];
    int n;
    while(scanf("%d",&n)!=EOF&&n>=1&&n<=100000)
    {
        int i;
    int x;
    for(i=0;i<n;i++)
    {
        scanf("%s",str);
        if(strcmp(str,"POP")==0)
        {
           if(top2==-1)
           {
               while(top1!=-1)
               {
                   b[++top2]=a[top1--];
               }
           }
           if(top2==-1)
            printf("-1\n");
           else
           {
               printf("%d\n",b[top2--]);
           }
        }
        else
        {
            scanf("%d",&x);
            a[++top1]=x;
        }
    }
    }

    return 0;
}