【2022初春】【LeetCode】232. 用栈实现队列

2022/2/1 23:11:20

本文主要是介绍【2022初春】【LeetCode】232. 用栈实现队列,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!

一遍过了,中间差一个判断

class MyQueue {
    Stack<Integer> a;
    Stack<Integer> b;
    public MyQueue() {
        a = new Stack<Integer>();
        b = new Stack<Integer>();
    }
    
    public void push(int x) {
        a.push(x);
        if(b.isEmpty()) b.push(x);
    }
    
    public int pop() {
        int ans = b.pop();
        if(b.isEmpty()&&!a.isEmpty()){
            while(!a.isEmpty()){
                b.push(a.peek());
                a.pop();
            }
            if(b.peek()==ans) b.pop();
        }
        return ans;
    }
    
    public int peek() {
        return b.peek();
    }
    
    public boolean empty() {
        if(a.isEmpty()&&b.isEmpty()) return true;
        else return false;
    }
}

/**
 * Your MyQueue object will be instantiated and called as such:
 * MyQueue obj = new MyQueue();
 * obj.push(x);
 * int param_2 = obj.pop();
 * int param_3 = obj.peek();
 * boolean param_4 = obj.empty();
 */


这篇关于【2022初春】【LeetCode】232. 用栈实现队列的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!


扫一扫关注最新编程教程