[刷题]Implement Queue by Two Stacks

最新推荐文章于 2025-07-29 14:24:32 发布

原创最新推荐文章于 2025-07-29 14:24:32 发布 · 275 阅读

0 ·

CC 4.0 BY-SA版权

data structure 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的方法。通过定义两个栈stack1和stack2，实现了队列的基本操作：push、pop和top。当进行push操作时，元素直接压入stack1；进行pop或top操作时，如果stack2为空，则将stack1中的所有元素依次弹出并压入stack2，最后返回stack2的弹出元素或顶部元素。

[LintCode]Implement Queue by Two Stacks

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

    public Solution() {
        // 2015-09-03
        stack1 = new Stack<>();
        stack2 = new Stack<>();
    }
    
    public void push(int element) {
        stack1.push(element);
    }

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

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