Implement Queue using Stacks

最新推荐文章于 2023-10-09 00:28:50 发布

原创最新推荐文章于 2023-10-09 00:28:50 发布 · 225 阅读

0 ·

CC 4.0 BY-SA版权

leetCode 专栏收录该内容

199 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过将入队操作放置在一个栈中，并在需要出队时将该栈的数据倒置到另一个栈来完成。这种方式能够有效地模拟队列的先进先出特性。

题目链接

思路：
两个栈。当有输出的时候。如果B里有东西就pop.没有的话把所有A里面的东西拿出来放入B。所有的入队全部放在A里面。

import java.util.Stack;

class MyQueue {
    // Push element x to the back of queue.

    Stack<Integer> stackA=new Stack<Integer>();
    Stack<Integer> stackB=new Stack<Integer>();
    boolean isOut=true;


    public void push(int x) {
        stackA.push(x);
    }

    // Removes the element from in front of queue.
    public void pop() {
        if(stackB.isEmpty())
        {
            while(!stackA.isEmpty())
            {
                stackB.push(stackA.pop());
            }
        }
        stackB.pop();
    }

    // Get the front element.
    public int peek() {
          if(stackB.isEmpty())
          {
            while(!stackA.isEmpty())
            {
                stackB.push(stackA.pop());
            }
          }
          return stackB.peek();
    }

    // Return whether the queue is empty.
    public boolean empty() {
        return stackA.isEmpty()&&stackB.isEmpty();
    }
}