代码随想录|232.用栈实现队列，225.用队列实现栈

栈与队列基础

首先我们需要了解什么是栈与队列。这是两种数据结构，可有底层的数据结构实现。
栈的特点是先进后出，类似碟盘子，一个个网上面放，并且取的时候从上面开始取。进出仅通过一个口子，也即口子既是头，也是尾。但是进出的方向不同。栈的基本操作有：

• push( x ): 将x元素加入栈尾
• pop(): 移除并返回栈尾元素
• peak(): 返回栈尾元素
• empty(): 栈为空，则返回TRUE， 非空返回FALSE

队列的特点是先进先出， 犹如直通的水管，从一个口子进，另一个口子出，进出是一个方向，但是头尾不是同一个口子。基本操作有：

• push(x): 将元素x加入队尾
• pop(): 移除并返回队列头部元素
• peak(): 返回队头元素
• empty(): 判断队列是否为空

用栈实现队列

题目链接leetcode 232.

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

由于队列与栈的最大区别在于头部不一样，队列有着头尾两个口，而栈只有一个。所以我们需要用两个栈分别实现队列的头尾功能，一个输入栈管理队尾的输入，一个输出栈管理队头的弹出或者查询。
队尾的输入与栈一致，只需要在输入栈里push元素即可。empty 功能只需要设置一个变量 size记录队列元素个数，并判断size 是否为零即可。主要的问题在于 pop() 的实现， peak() 与 pop() 相似，只是没有移除元素。
如何实现pop() 队头的功能呢？注意到将输入栈的元素全部输入输出栈，那么输入栈的头部（也即队列的头部）则变成了输出栈的尾部，这样pop() 或者 peak() 输出栈即可实现队头的弹出或者查询。
一个朴素的想法就是每当需要pop 或者peak的时候，将输入栈全部转入输出栈，然后pop，peak输出栈栈尾，最后将输出栈在全部转回输入栈。但是显然这样的操作会浪费一些时间，而且输出栈大部分时间是空的。
一种优化的思想是：当需要pop的时候，先判断输出栈是否为空，若不为空，直接pop；若为空，则将输入栈全部转入输出栈，然后再进行pop。

Python 代码实现

在python标准库中，有collections.deque 双端队列可以实现栈。这里用list来实现栈的功能，其中

• append() 对应 push()
• pop() 仍然可用

peak由列表pop之后再加回去。

class MyQueue:

    def __init__(self):
        self.sta_in = []
        self.sta_out = []
        self.size = 0

    def push(self, x: int) -> None:
        self.sta_in.append(x)
        self.size += 1

    def pop(self) -> int:
        if self.empty():
            return None
        if len(self.sta_out) == 0:
            self.shift()
        self.size -= 1
        return self.sta_out.pop()

    def peek(self) -> int:
        if self.empty():
            return None
        if len(self.sta_out) == 0:
            self.shift()
        ans = self.sta_out.pop()
        self.sta_out.append(ans)
        return ans

    def empty(self) -> bool:
        return self.size == 0
	# shift 函数将输入栈全部转入输出栈
    def shift(self) -> None:
        while (len(self.sta_in)):
            self.sta_out.append(self.sta_in.pop())
        return 
        
# Your MyQueue object will be instantiated and called as such:
# obj = MyQueue()
# obj.push(x)
# param_2 = obj.pop()
# param_3 = obj.peek()
# param_4 = obj.empty()

用队列实现栈

题目链接 leetcode

题目描述： 请你用一个队列实现栈的全部四种操作（push、top、pop 和 empty）

push,empty的方法与栈的差不多，我们一样考虑pop。如何将队列的头变成栈的尾呢？ 注意到将队列的头部连接到队尾，即将元素一个个弹出并添加到队尾，直至需要弹出的元素到了队头，这次只弹出，不添加，这样我们就实现了队尾的弹出。

Python 代码实现

队列的pop()将有list.pop(0)实现。

class MyStack:

    def __init__(self):
        self.que = []
        self.size = 0

    def push(self, x: int) -> None:
        self.que.append(x)
        self.size += 1

    def pop(self) -> int:
        self.circle()
        self.size -= 1
        return self.que.pop(0)

    def top(self) -> int:
        self.circle()
        ans = self.que.pop(0)
        self.que.append(ans)
        return ans

    def empty(self) -> bool:
        return self.size == 0

    def circle(self) -> None:
        for _ in range(self.size -1):
            self.que.append(self.que.pop(0))
        
# 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()