代码随想录算法训练营第十天（py）|栈与队列| 232.用栈实现队列、225. 用队列实现栈

232. 用栈实现队列（easy）

力扣链接

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

实现 MyQueue 类：

void push(int x) 将元素 x 推到队列的末尾
int pop() 从队列的开头移除并返回元素
int peek() 返回队列开头的元素
boolean empty() 如果队列为空，返回 true ；否则，返回 false

思路

使用栈来模拟队列的难点在于如何用 先进后出 来模拟 先进先出 ，对于puch可以直接用append来实现，对于empty更是简单，难点在于pop和peek的实现。

1. 构建两个空栈stack_in和stack_out，前者负责输入工作，后者负责输出工作。
2. puch时，直接将元素append到stack_in中。
3. pop时，期望得到stack_in[0]，如果直接用stack_in.pop()会返回stack_in[-1]。只需要先将stack_in依次pop到stack_out中，此时stack_in[0]就跑到了stack_out[-1]的位置，现在再stack_out.pop()即可得到答案。
4. peek时，只需要pop一下，再把pop出去的值append到stack_out中

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: # 如果输出栈中有元素，就直接pop
            return self.stack_out.pop()
        else: # 如果没有，就把输入栈中所有元素全部扔到输出栈里，然后pop
            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:
        ret = self.pop()
        self.stack_out.append(ret)
        return ret


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

225. 用队列实现栈（easy）

请你仅使用两个队列实现一个后入先出（LIFO）的栈，并支持普通栈的全部四种操作（push、top、pop 和 empty）。

实现 MyStack 类：

void push(int x) 将元素 x 压入栈顶。
int pop() 移除并返回栈顶元素。
int top() 返回栈顶元素。
boolean empty() 如果栈是空的，返回 true ；否则，返回 false 。

思路1 直接用数组得了

class MyStack:

    def __init__(self):
        self.list = []

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

    def pop(self) -> int:
        return self.list.pop()

    def top(self) -> int:
        ans = self.list.pop()
        self.list.append(ans)
        return ans

    def empty(self) -> bool:
        return not self.list

思路2 老老实实用队列

class MyStack:

    def __init__(self):
        self.queue1 = collections.deque()
        self.queue2 = collections.deque()


    def push(self, x: int) -> None:
        self.queue2.append(x)
        while self.queue1:
            self.queue2.append(self.queue1.popleft())
        self.queue1, self.queue2 = self.queue2, self.queue1


    def pop(self) -> int:
        return self.queue1.popleft()


    def top(self) -> int:
        return self.queue1[0]


    def empty(self) -> bool:
        return not self.queue1