剑指Offer笔记:图解09. 用两个栈实现队列 30. 包含min函数的栈

目录

python题解:

list常见使用方法: 

python题解:

参考题解:


剑指 Offer 09. 用两个栈实现队列

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

示例 1:

输入:
["CQueue","appendTail","deleteHead","deleteHead","deleteHead"]
[[],[3],[],[],[]]
输出:[null,null,3,-1,-1]

示例 2:

输入:
["CQueue","deleteHead","appendTail","appendTail","deleteHead","deleteHead"]
[[],[],[5],[2],[],[]]
输出:[null,-1,null,null,5,2]

3733d263a73c44d08cc95efde4a3fa90.png

 通过设置辅助栈来实现队列,每当删除队列元素时,都会先把A栈元素依次pop()并push()进B栈,实现倒序,从而B中元素顺序和队列实现相同。

将一列元素颠倒两次(两个栈)实现队列先进先出

python题解:

class CQueue:

    def __init__(self): # 用于创建实例时初始化
        self.A=[]
        self.B=[]

    def appendTail(self, value: int) -> None: # 加入元素(push)进入A栈
        self.A.append(value)

    def deleteHead(self) -> int:  # pop元素出栈
        if self.B: return self.B.pop()  # 如果B栈不为空,就直接pop栈顶元素
        if self.A:      #B栈为空,A栈不为空
            while self.A:
                self.B.append(self.A.pop())  # B栈为空的时候,循环将A栈中的元素push进B栈
            return self.B.pop()
        else: return -1   # B栈为空,A栈也为空


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

list常见使用方法: 

642f6983dad24def9ff011387e13d69c.png

剑指 Offer 30. 包含min函数的栈

定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。

示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min();   --> 返回 -3.
minStack.pop();
minStack.top();      --> 返回 0.
minStack.min();   --> 返回 -2.

 0991b1cff41546999f02befae1e31658.png

 同样使用双栈解决问题,B栈(辅助栈)用来存储当前实际栈中可能成为最小值的元素,B栈栈顶元素即为最小元素。

min()  top()只返回数值,并不会改变栈的状态

python题解:

class MinStack:

    def __init__(self):    #初始化
       self.A,self.B=[],[]


    def push(self, x: int) -> None:   
        self.A.append(x)                 # 先把元素push进A栈
        if not self.B or x<=self.B[-1]:  #判断当B栈为空或者此时进栈的元素比当前B栈栈顶的元素 
                                         #值小的时候,把该元素也push进B栈
            self.B.append(x)

    def pop(self) -> None:
        if self.A.pop()==self.B[-1]:   #如果要pop的值在A、B栈栈顶(当前最小值),就都pop出去
            self.B.pop()

    def top(self) -> int:
        return self.A[-1]  # 返回A栈栈顶元素

    def min(self) -> int:
        return self.B[-1]  # 返回B栈栈顶元素,即最小值


# Your MinStack object will be instantiated and called as such:
# obj = MinStack()
# obj.push(x)
# obj.pop()
# param_3 = obj.top()
# param_4 = obj.min()

参考题解:

jyd - 力扣(LeetCode)

Python3 列表 | 菜鸟教程 (runoob.com)

猜你喜欢

转载自blog.csdn.net/qq_56864896/article/details/130138846