(力扣每日一题)用两个栈实现队列

用两个栈实现队列

用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )
解题思路
1、栈无法实现队列功能: 栈底元素(对应队首元素)无法直接删除,需要将上方所有元素出栈。
2、双栈可实现列表倒序: 设有含三个元素的栈 A = [1,2,3]和空栈 B = []。若循环执行 A 元素出栈并添加入栈 B ,直到栈 A 为空,则 A=[] , B=[3,2,1] ,即 栈 B元素实现栈 A元素倒序 。
3、利用栈 B 删除队首元素: 倒序后,B执行出栈则相当于删除了 A 的栈底元素,即对应队首元素。
4、只需要求实现 加入队尾appendTail() 和 删除队首deleteHead() 两个函数的正常工作,因此我们可以设计栈 A 用于加入队尾操作,栈 B 用于将元素倒序,从而实现删除队首元素。
代码

class CQueue:
    def __init__(self):
        # 设定两个栈,一个负责入,一个负责出
        # 栈顶为队列的尾
        self.in_stack = [] 
        # 栈顶为队列的首
        self.out_stack = []  

    def appendTail(self, value: int) -> None:
        # in_stack只负责模拟进入队列操作
        self.in_stack.append(value)

    def deleteHead(self) -> int:
        # out_stack 负责模拟出队列的操作
        # 若out_stack为空
        if not self.out_stack: 
            # 如果in_stack也为空,说明队列为空,按题意直接返回-1
            if not self.in_stack: 
                return -1
# out_stack空,in_stack不空,说明队列列还有元素存在,就先将in_stack中的导入out_stack中,再弹出out_stack中最后一个    
            else:  
                while self.in_stack:
                    self.out_stack.append(self.in_stack.pop())
                return self.out_stack.pop()
        else:
            return self.out_stack.pop()

# Your CQueue object will be instantiated and called as such:
# obj = CQueue()
# obj.appendTail(value)
# param_2 = obj.deleteHead()

你可能感兴趣的:(常见算法题归纳)