【Day 5 】232. 用栈实现队列

题目地址(232. 用栈实现队列)

https://leetcode-cn.com/problems/implement-queue-using-stacks/

题目描述

使用栈实现队列的下列操作：

push(x) -- 将一个元素放入队列的尾部。
pop() -- 从队列首部移除元素。
peek() -- 返回队列首部的元素。
empty() -- 返回队列是否为空。
示例:

MyQueue queue = new MyQueue();

queue.push(1);
queue.push(2);
queue.peek(); // 返回 1
queue.pop(); // 返回 1
queue.empty(); // 返回 false
说明:

你只能使用标准的栈操作 -- 也就是只有 push to top, peek/pop from top, size, 和 is empty 操作是合法的。
你所使用的语言也许不支持栈。你可以使用 list 或者 deque（双端队列）来模拟一个栈，只要是标准的栈操作即可。
假设所有操作都是有效的、 （例如，一个空的队列不会调用 pop 或者 peek 操作）。

编程语言

python3

思路

0. 考察栈、入栈、出栈、用栈模拟队列
1. 解法一：用pop(0)直接弹出第一个数  
2. 解法二：辅助栈

代码

解法一：
class MyQueue:
    def __init__(self):
        self.queue = []

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

    def pop(self) -> int:
        first_pop = self.queue.pop(0)
        return first_pop

    def peek(self) -> int:
        first = self.queue[0]
        return first

    def empty(self) -> bool: 
        return self.queue == []
 解法二： 
class MyQueue:
    def __init__(self): 
        self.instack = []
        self.outstack = []

    def in2out(self):
        while (self.instack != []):
            self.outstack.append(self.instack.pop())
        
    def push(self, x: int) -> None:
        self.instack.append(x)

    def pop(self) -> int:
        if self.outstack == []:
            self.in2out()
        return self.outstack.pop()

    def peek(self) -> int:
        if self.outstack == []:
            self.in2out()
        return self.outstack[-1]

    def empty(self) -> bool:
        return (self.outstack==[]) & (self.instack==[])

复杂度分析

0. 解法一
时间复杂度：O(1)
空间复杂度：O(n)
1. 解法二
时间复杂度：O(1)
空间复杂度：O(n)

收获

0. 若array为空则输出1 不为空输出0
   不要用len和if else和==(耗时更多)！！！
   用 return self.array == [] 这样写耗时少且简洁

1. 辅助栈完成队列