代码随想录算法训练营10day-python|232. 用栈实现队列、225. 用队列实现栈

一、232. 用栈实现队列

思路：队列是一种先入先出的数据结构，而栈是后入先出的数据结构，所以这里需要两个栈，一个输入栈一个输出栈。

比如：队列 1 2 3 4进，出应该是 1 2 3 4，但是由于栈是后入先出，出来只能是4 3 2 1，和队列是相反的，所以就再需要一个栈，把4 3 2 1 返成 1 2 3 4。输入栈来反转元素的入队顺序，元素入只能从输入栈进（push）。输出栈用来存储元素的正常顺序，元素出只能从输出栈出（pop，peek）。

总结： emo了几天终于出总结啦（其实是忙飞了，累到走路能睡着）。

push的操作就是直接入栈，pop出队操作，如果输出栈不为空的话，输出栈顶元素，为空的话就把输入栈的所有元素压入输出栈中，然后再输出栈顶元素。peek 取队列的首元素， pop 的操作差不多。区别是pop 删掉队首元素，peek 是看队首元素。empty 做判空操作，输入输出栈都为空则队列为空。

代码：双栈

class MyQueue:

    def __init__(self):
        # 初始化输入输出栈
        self.stack_in = []
        self.stack_out = []

    def push(self, x: int) -> None:
        # 有新元素，进入输入栈
        self.stack_in.append(x)

    def pop(self) -> int:
        # 如果为空
        if self.empty():
            return None
        # 如果输出栈不为空，返回输出栈元素
        if self.stack_out:
            return self.stack_out.pop()
        # 输出栈为空，将输入的元素压到输出栈
        else:
            for i in range(len(self.stack_in)):
                self.stack_out.append(self.stack_in.pop())
            return self.stack_out.pop()

    def peek(self) -> int:
        # 调用已有函数pop，复用
        ans = self.pop()
        # pop函数弹出ans，所以要添加回去
        self.stack_out.append(ans)
        return ans

    def empty(self) -> bool:
        return not (self.stack_in or self.stack_out)

二、225. 用队列实现栈

思路：这道题跟上面的思路差不多，建一个主队列，其余操作正常，每次模拟栈弹出元素的时候，将除了最后一个元素外的之前元素弹出队列，重新添加到主队列尾部即可。

总结：视频讲解+画图，就比较的通俗易懂，加上卡哥的视频讲得比较细，适合我这种小白

代码：

from collections import deque

class MyStack:

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

    def push(self, x: int) -> None:
        self.queue_in.append(x)

    def pop(self) -> int:
        # 确认empty是否为空
        if self.empty():
            return None
        # 不为空则把in的所有元素（除最后一个）依次出列放到out中
        for i in range(len(self.queue_in) - 1):
            self.queue_out.append(self.queue_in.popleft())
        # 交换 in out内容，此时out只有一个元素
        self.queue_in, self.queue_out = self.queue_out, self.queue_in
        return self.queue_out.popleft()

    def top(self) -> int:
        if self.empty():
            return None
        for i in range(len(self.queue_in) - 1):
            self.queue_out.append(self.queue_in.popleft())
        self.queue_in, self.queue_out = self.queue_out, self.queue_in
        # 把out中的pop出来，即是最后一个
        tmp = self.queue_out.popleft()
        # 追加tmp到in的末尾
        self.queue_in.append(tmp)
        return tmp

    def empty(self) -> bool:
        # 只有in有数据，只判断in是否为空
        return len(self.queue_in) == 0