【20191217】【每天一道算法题】用队列实现栈（双队列）

本文链接：https://blog.csdn.net/weixin_40583722/article/details/103579223

问题

使用队列实现栈的下列操作：

push(x) -- 元素 x 入栈
pop() -- 移除栈顶元素
top() -- 获取栈顶元素
empty() -- 返回栈是否为空

注意:

你只能使用队列的基本操作-- 也就是 push to back, peek/pop from front, size, 和 is empty 这些操作是合法的。

你所使用的语言也许不支持队列。你可以使用 list 或者 deque（双端队列）来模拟一个队列 , 只要是标准的队列操作即可。

你可以假设所有操作都是有效的（例如, 对一个空的栈不会调用 pop 或者 top 操作）。

思路及代码

# 方法一：双队列。 
# 思路：使用queue1存数据，queue2辅助实现先进先出的功能。
class MyStack:

    def __init__(self):
        """
        Initialize your data structure here.
        """
        self.queue1 = []   # 存结果
        self.queue2 = []   # 辅助

    def push(self, x: int) -> None:
        """
        Push element x onto stack.
        """
        self.queue1.append(x)

    def pop(self) -> int:
        """
        Removes the element on top of the stack and returns that element.
        """
        while self.queue1:
            self.queue2.append(self.queue1.pop(0))   # L.pop(idx)：移除列表的第idx个元素（默认为最后一个）
        tmp = self.queue2.pop()
        while self.queue2:
            self.queue1.append(self.queue2.pop(0)) 
        return tmp

    def top(self) -> int:
        """
        Get the top element.
        """
        while self.queue1:
            self.queue2.append(self.queue1.pop(0))
        tmp = self.queue2.pop()
        self.queue2.append(tmp)
        while self.queue2:
            self.queue1.append(self.queue2.pop(0))
        return tmp

    def empty(self) -> bool:
        """
        Returns whether the stack is empty.
        """
        return self.queue1 == []


# Your MyStack object will be instantiated and called as such:
# obj = MyStack()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.top()
# param_4 = obj.empty()