算法系列------利用栈实现队列

原创于 2019-10-14 22:02:11 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

每日一算专栏收录该内容

23 篇文章

订阅专栏

本文介绍了一种使用两个栈来实现队列的数据结构方法。通过将输入栈用于元素的添加，输出栈用于元素的移除，实现了队列的先进先出特性。当输出栈为空时，所有元素从输入栈转移到输出栈，以确保正确性。

class MyQueue {
    private Stack<Integer> a;// 输入栈
    private Stack<Integer> b;// 输出栈
    
    public MyQueue() {
        a = new Stack<>();
        b = new Stack<>();
    }
    
    public void push(int x) {
        a.push(x);
    }
    
    public int pop() {
        // 如果b栈为空，则将a栈全部弹出并压入b栈中，然后b.pop()
        if(b.isEmpty()){
            while(!a.isEmpty()){
                b.push(a.pop());
            }
        }
        return b.pop();
    }
    
    public int peek() {
        if(b.isEmpty()){
            while(!a.isEmpty()){
                b.push(a.pop());
            }
        }
        return b.peek();
    }
    
    public boolean empty() {
        return a.isEmpty() && b.isEmpty();
    }
}