python实现【用两个栈实现队列】力扣题

题目:用两个栈实现队列
具体描述
用两个栈实现一个队列。队列的声明如下,请实现它的两个函数 appendTail 和 deleteHead ,分别完成在队列尾部插入整数和在队列头部删除整数的功能。(若队列中没有元素,deleteHead 操作返回 -1 )

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/yong-liang-ge-zhan-shi-xian-dui-lie-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
题解前若无思路可看这里漫画更清晰
题解

class CQueue:

    def __init__(self):
        self.stack1=[] #基本栈
        self.stack2=[] #辅助栈


    def appendTail(self, value: int) -> None:
        self.stack1.append(value) #元素放入基本栈


    def deleteHead(self) -> int:
        # 将基本栈的元素放入辅助栈,基本栈后进元素先放入辅助栈
        if len(self.stack2) == 0:
            while len(self.stack1) != 0:
                self.stack2.append(self.stack1.pop())
        # 从辅助栈放出元素
        if len(self.stack2) != 0:
            return(self.stack2.pop())
        return(-1)

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

复杂度分析

(1)插入元素
时间复杂度:O(1)。向基本栈插入一个元素,在队列尾部插入元素,使用常数时间。
空间复杂度:O(1)。向基本栈插入一个元素,使用常数空间。
(2)删除元素
时间复杂度:O(n)。基本栈的元素要删除并插入到辅助栈,是线性时间复杂度。
空间复杂度:O(n)。需要额外空间储存元素。
力扣官方给的题解插入删除元素复杂度正好和上述相反,其将栈元素删除压入至辅助栈放入appendTail插入元素步,个人理解中,对于删除操作少的,插入元素操作多的,将appendTail更简单更好一些

你可能感兴趣的:(力扣)