【leetcode刷刷】232.用栈实现队列、225. 用队列实现栈

232. 用栈实现队列

1. 维护两个栈，stack_in和stack_out。in负责push，out负责pop。
2. pop的逻辑是，如果out里有东西，则直接从out中pop(-1)；直到out里空了，再把in里的转移到out里，再继续从out中pop(-1)。
3. 需要注意的是，peek里，return的是pop的结果，也就是队列的头。但是要加回去的时候，加在stack_out里。因为是从out里pop出去的。

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(-1)
        else:
            while self.stack_in:              
                self.stack_out.append(self.stack_in.pop(-1))
            x = self.stack_out.pop(-1)
            return x


    def peek(self) -> int:
        ans = self.pop()
        self.stack_out.append(ans)
        return ans

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



# 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()

225. 用队列实现栈

1. 用队列实现栈，和用栈实现队列不一样。因为队列先入先出，倒入另一个队列也先入先出；但是一个栈先入后出，倒入另一个栈就是先入先出了。
2. 因此事实上只需要一个队列，通过计算队列的长度，来取出最后一个入队的数，比较暴力啦。另一个队列是用来备份的。

class MyStack:

    def __init__(self):
        self.q_in = []
        # self.q_out = []
        self.q_out2 = []


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

    # 先入后出
    def pop(self) -> int:
        if self.empty():
            return 
        
        for i in range(len(self.q_in) - 1):
            self.q_out2.append(self.q_in.pop(0))
        self.q_in, self.q_out2 = self.q_out2, self.q_in    
        return self.q_out2.pop(0)
        


    def top(self) -> int:
        if self.empty():
            return 
        
        x = self.pop()
        self.q_in.append(x)
        return x


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



# 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()