【力扣】232. 用栈实现队列

以下为力扣官方题解

题目

请你仅使用两个栈实现先入先出队列。队列应当支持一般队列的支持的所有操作（$push、pop、peek、empty）$：
实现 $MyQueue$ 类：

• $voidpush(intx)$ 将元素 $x$ 推到队列的末尾
• $intpop()$ 从队列的开头移除并返回元素
• $intpeek()$ 返回队列开头的元素
• $booleanempty()$ 如果队列为空，返回 $true$ ；否则，返回 $false$

说明：

• 你只能使用标准的栈操作 —— 也就是只有 $pushtotop,peek/popfromtop,size,和isempty$ 操作是合法的。
• 你所使用的语言也许不支持栈。你可以使用 $list$ 或者 $deque$（双端队列）来模拟一个栈，只要是标准的栈操作即可。

进阶：

你能否实现每个操作均摊时间复杂度为 $O(1)$ 的队列？换句话说，执行 $n$ 个操作的总时间复杂度为 $O(n)$ ，即使其中一个操作可能花费较长时间。

示例：

输入：
$["MyQueue","push","push","peek","pop","empty"]$
$[[],[1],[2],[],[],[]]$
输出：
$[null,null,null,1,1,false]$
解释：
$MyQueuemyQueue=newMyQueue();$
$myQueue.push(1);//queueis:[1]$
$myQueue.push(2);//queueis:[1,2](leftmostisfrontofthequeue)$
$myQueue.peek();//return1$
$myQueue.pop();//return1,queueis[2]$
$myQueue.empty();//returnfalse$

提示：

• $1<=x<=9$
• 最多调用 $100$ 次 $push、pop、peek$ 和 $empty$
• 假设所有操作都是有效的 （例如，一个空的队列不会调用 $pop$ 或者 $peek$ 操作）

官方题解

方法一：双栈

思路

将一个栈当作输入栈，用于压入 $push$ 传入的数据；另一个栈当作输出栈，用于 $pop$ 和 $peek$ 操作。

每次 $pop$ 或 $peek$ 时，若输出栈为空则将输入栈的全部数据依次弹出并压入输出栈，这样输出栈从栈顶往栈底的顺序就是队列从队首往队尾的顺序。

代码

class MyQueue {
    Deque<Integer> inStack;
    Deque<Integer> outStack;

    /** Initialize your data structure here. */
    public MyQueue() {
        inStack = new LinkedList<Integer>();
        outStack = new LinkedList<Integer>();
    }
    
    /** Push element x to the back of queue. */
    public void push(int x) {
        inStack.push(x);
    }
    
    /** Removes the element from in front of queue and returns that element. */
    public int pop() {
        if (outStack.isEmpty())
        {
            in2out();
        }
        return outStack.pop();
    }
    
    /** Get the front element. */
    public int peek() {
        if (outStack.isEmpty())
        {
            in2out();
        }
        return outStack.peek();
    }
    
    /** Returns whether the queue is empty. */
    public boolean empty() {
        return inStack.isEmpty() && outStack.isEmpty();
    }

    private void in2out()
    {
        while (!inStack.isEmpty())
        {
            outStack.push(inStack.pop());
        }
    }
}

/**
 * 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();
 */

复杂度分析

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