LeetCode.232. 用栈实现队列

LeetCode.232. 用栈实现队列

难度：easy

 

 

 运用两个栈来实现一个队列，一个为入栈一个为出栈。

class MyQueue {

    Stack<Integer> stackIn;
    Stack<Integer> stackOut;

    public MyQueue() {
        stackIn = new Stack<>();
        stackOut = new Stack<>();
    }
    
    public void push(int x) {
        stackIn.push(x);
    }
    
    public int pop() {
        if (stackOut.isEmpty()) {
            while (!stackIn.isEmpty()) {
                stackOut.push(stackIn.pop());
            }
        } 
        return stackOut.pop();
    }
    
    public int peek() {
        if (stackOut.isEmpty()) {
            while (!stackIn.isEmpty()) {
                stackOut.push(stackIn.pop());
            }
        } 
        return stackOut.peek();
    }
    
    public boolean empty() {
        if (stackIn.isEmpty() && stackOut.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();
 */

复杂度分析

• 时间复杂度：O（1），push 和 empty 为 O(1)，pop 和peek 为均摊O(1)。对于每个元素，至多入栈和出栈各两次，故均摊复杂度为 O(1)。
• 空间复杂度：O(n)。其中 nn 是操作总数。对于有 n 次push 操作的情况，队列中会有 n个元素，故空间复杂度为O(n)。