代码随想录第十天：232.用栈实现队列、225. 用队列实现栈

232.用栈实现队列

要通过画图来模拟弹出的过程

class MyQueue(object):

    def __init__(self):
        self.stack_push = []  # 一个栈用来放入元素
        self.stack_pop = []    # 另一个用来弹出元素，实现先进先出

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.stack_push.append(x)


    def pop(self):
        """
        :rtype: int
        """
        if self.stack_pop:
            return self.stack_pop.pop()
        else:
            for i in range(len(self.stack_push)):  # 栈只能后进先出，因此要将push栈中的元素pop()后装入pop栈，这样弹出的顺序就是对的（画图）
                self.stack_pop.append(self.stack_push.pop())
            return self.stack_pop.pop()


    def peek(self):
        """
        :rtype: int  # 就是返回第一个要弹出的元素
        """
        res = self.pop()  # 调用上面一个函数
        self.stack_pop.append(res)  # 找到该元素的同时也弹出了，所以要再塞回去
        return res



    def empty(self):
        """
        :rtype: bool
        """
        return self.stack_push == [] and self.stack_pop == []  # 当两个栈都为空时，该队列为空

225.用队列实现栈​​​​​​

from collections import deque
# 双端队列，只使用append和popleft
class MyStack(object):

    def __init__(self):
        self.queue_in = deque()
        self.queue_out = deque()

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.queue_in.append(x)


    def pop(self):
        """
        :rtype: int
        """
        if self.empty():
            return None
        for i in range(len(self.queue_in)-1): # 将栈顶元素下面的所有元素都先备份到out队列中，
            self.queue_out.append(self.queue_in.popleft())
        res = self.queue_in.pop()  # 只剩下栈顶元素
        for i in range(len(self.queue_out)):  # 将out队列中备份的元素还放回原来的in队列中
            self.queue_in.append(self.queue_out.popleft())
        return res


    def top(self):
        """
        :rtype: int
        """
        res = self.pop()
        self.queue_in.append(res)
        return res



    def empty(self):
        """
        :rtype: bool
        """
        return not self.queue_in

刚开始写这道题的pop时，直接用了list来实现，然后直接pop(0)，但是其实列表的pop(0)时间复杂度为O(n)，所以用collections.deque双端队列，时间复杂度为O(1)，同时要求用队列来实现栈，所以只用deque中的append和popleft；

设置两个双端队列in和out，in用来存数据，out用来需要的时候备份，这题不像上一题，上一题中将数据放到另外一个栈中顺序就对了，可以直接弹出，这题两个都是先进先出，所以得将除了栈顶元素之外的其他元素先移走；

最后我将out队列中备份的元素还放回原来的in队列中，题解中是将in和out交换位置

题解：代码随想录

代码随想录