代码随想录算法训练营第10天|栈和队列 LeetCode 232.用栈实现队列、225. 用队列实现栈

• 栈：先入后出
• 队列：先进先出

232.用栈实现队列

• 用两个栈来实现队列
• 当需要出队操作时，必须把入栈中的所有元素压入出栈中，这样才能改变所有元素的顺序，不能有元素滞留，不然顺序就乱了。但是在压之前，需要先判断出栈是否为空，如果不为空，直接pop出栈中的元素，如果为空，就将入栈中的所有元素压入出栈中。如果在出栈不为空的状态下压入元素，同样也会导致顺序错乱。
• 判断队列是否为空：当出栈和入栈都为空时，队列为空

class MyQueue(object):

    def __init__(self):
        self.stack_in = []
        self.stack_out = []
        

    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.stack_in.append(x)
       
        
    def pop(self):
        """
        :rtype: int
        """
        if self.stack_out != []:
            return self.stack_out.pop()
        else: # 如果出栈为空，将入栈中的所有元素都压入出栈中
            while self.stack_in != []:
                self.stack_out.append(self.stack_in.pop())
            return self.stack_out.pop()


    def peek(self):
        """
        :rtype: int
        """
        # 跟pop的功能相似，只是不让元素出栈，直接调用pop函数
        res = self.pop()
        self.stack_out.append(res)   # 取出栈顶后还要再压进去
        return res
        

    def empty(self):
        """
        :rtype: bool
        """
        if self.stack_in == [] and self.stack_out == []:
           return True
        else:
            return False

# 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. 用队列实现栈

• 队列是先进先出，不管用几个队列，都不能像之前的栈一样改变元素之间的顺序。
• 用一个队列实现栈
• 队列是先进先出，而栈是先进后出，栈的弹出操作其实是弹出队列的最后一个元素，那么怎么做到呢？其实就是让前面的元素一个个弹出再重新加入到队列的尾部，重复这个操作n-1次，直到最后一个元素被排到队首。（两个队列的话，就是把前n-1个元素从que1挪到que2中，然后从que1弹出最后一个元素，再把que2中的元素挪回到que1中）

from collections import deque

class MyStack(object):

    def __init__(self):
        self.que = []
        
    def push(self, x):
        """
        :type x: int
        :rtype: None
        """
        self.que.append(x)
        

    def pop(self):
        """
        :rtype: int
        """
        # 弹出栈顶，即队尾，依次把前n-1个元素弹出再重新加入到队列中
        n = len(self.que)
        n -= 1
        while n > 0:
            self.que.append(self.que.pop(0))
            n -= 1
        return self.que.pop(0)
        

    def top(self):
        """
        :rtype: int
        """
        return self.que[-1]
        
    def empty(self):
        """
        :rtype: bool
        """
        return self.que == []



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