[Lintcode]Implement Queue by Two Stacks

最新推荐文章于 2023-07-29 01:49:56 发布

青铁

最新推荐文章于 2023-07-29 01:49:56 发布

阅读量208

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： lintcode

本文链接：https://blog.youkuaiyun.com/jc69186918/article/details/52750552

算法专栏收录该内容

191 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过这种方式，可以有效地支持队列的基本操作，如push、pop和top。当执行pop或top操作时，如果出栈为空，则将入栈的所有元素依次移至出栈。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

As the title described, you should only use two stacks to implement a queue's actions.

The queue should support push(element), pop() and top()where pop is pop the first(a.k.a front) element in the queue.

Both pop and top methods should return the value of first element.

解法：双stack，一个进一个出。Empty时不需要单独处理，抛出stack抛出的异常即可

public class Queue {
    private Stack<Integer> stack1;
    private Stack<Integer> stack2;

    public Queue() {
        //out
       stack1 = new Stack<Integer>();
       //in
       stack2 = new Stack<Integer>();
    }
    
    public void push(int element) {
        stack2.push(element);
    }

    public int pop() {
        if(stack1.isEmpty()) {
            while(!stack2.isEmpty()) stack1.push(stack2.pop());
        } 
        return stack1.pop();
    }

    public int top() {
        if(stack1.isEmpty()) {
            while(!stack2.isEmpty()) stack1.push(stack2.pop());
        } 
        return stack1.peek();
    }
}