代码随想录算法训练营第九天|232.用栈实现队列，225.用队列实现栈

文档参考：programmercarl.com

今天的题估计算开胃小菜吧，主要还是复习了一下栈和队列的使用，两个题实现栈和队列的互相构建还挺有意思

225.用栈实现队列

"""
用两个栈，一个in栈一个out栈来模拟队列
in栈负责保存队列进来的顺序，out栈是根据保存顺序pop出正确的值
"""

class MyQueue(object):
    def __init__(self):
        # 建立两个列表模拟栈
        self.stack_in = []
        self.stack_out = []

    def push(self, x):
        # 向in栈里面添加数据
        self.stack_in.append(x)

    def pop(self):
        if self.empty(): # 如果in栈为空，说明无法pop
            return None

        if self.stack_out:# 如果stack_out栈不为空，则直接pop出最后的那个就行
            return self.stack_out.pop()
        else: # out栈为空，则将in栈里的元素以反方向填充到out栈中，再弹出out栈最后那个元素
            for i in range(len(self.stack_in)):
                self.stack_out.append(self.stack_in.pop())
            return self.stack_out.pop()

    def peek(self):
        # 因为只要获取队列开头的值就行，所以需要添加回去
        the_first = self.pop()
        self.stack_out.append(the_first)
        return the_first

    def empty(self):
        # 只要in或out不是非空列表，就说明队列中还有元素
        if self.stack_out == [] and self.stack_in == []:
            return True
        else:
            return False

232.用栈实现队列

"""
用一个队列就可以实现栈
"""
from collections import deque

class MyStack(object):

    def __init__(self):
        # 新建一个队列
        self.queue = deque()

    def push(self, x):

        self.queue.append(x)

    def pop(self):

        # 判断是否为空
        if self.empty():
            return None
        # 依次 “弹出最先进入队列的元素，并再次将其添加回队列“
        for i in range(len(self.queue) - 1):
            self.queue.append(self.queue.popleft())
        return self.queue.popleft() # len - 1保证了现在队列头元素为原本队列中最末尾的元素

    def top(self):

        # 先记录栈顶元素
        temp = self.queue.pop()
        # 再添加回去
        self.queue.append(temp)
        return temp

    def empty(self):
 
        if self.queue != None:
            return False
        return True

轻轻松松，去玩p5r了~