2022跟着leedcode学数据结构--第31天

99 阅读2分钟

「这是我参与2022首次更文挑战的第31天,活动详情查看:2022首次更文挑战

image.png

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

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

 

示例 1:

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

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

1 <= values <= 10000 最多会对 appendTail、deleteHead 进行 10000 次调用

解题思路:

  • 本题考察栈和队列
  • 栈:后进先出
  • 队列:先进先出
  • 先初始化 两个栈 add_stack 和delete_stack add_stack 作为新增使用,delete_stack作为辅助作用
  • 新增的实现类似列表,直接append操作即可
  • delete操作 先判断add_stack是否为空 为空返回-1 判断delete_stack 是否存在,存在则进行出栈操作得到就是结果,否则就进行add_stack的出栈操作,将数据全部导入到delete_stack 然后对delete_stack做出栈操作得到的第一个数就是要的结果
class CQueue:

    def __init__(self):
        self.add_stack = []  # 入栈
        self.delete_stack = [] # 出栈


    def appendTail(self, value: int) -> None:
        self.add_stack.append(value)


    def deleteHead(self) -> int:
        if not self.add_stack:
            return -1
        if self.delete_stack:
            return self.delete_stack.pop()
        while self.add_stack:
            self.delete_stack.append(self.add_stack.pop())
        return self.delete_stack.pop()


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

执行结果:

image.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.  

提示:

各函数的调用总次数不超过 20000 次

解题思路:

  • 本题考察栈的使用
  • 使用两个栈一个用于存储所有值,一个用于存储最小值
  • 先初始化两个栈 stack 和 minStack
  • push操作 stack直接使用append方法即可,minStack 判断最后一个值比当前插入的值大,如果插入值更小,则进行入栈操作
  • pop 操作就是删除操作,在删除stack同时要注意删除minStack 中最最小值,如果和最小值相同则也要删除
  • top 操作就是直接获取stack[-1]得到的就是需要的值
  • min 获取最小值 minStack 中 minStack[-1]我们存储的就是最小值,只需要取出即可
class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.stack = []
        self.minStack=[]

    def push(self, x: int) -> None:
        self.stack.append(x)
        if not self.minStack:
            self.minStack.append(x)
        elif self.minStack and x <= self.minStack[-1]:
            self.minStack.append(x)


    def pop(self) -> None:
        tmp = self.stack.pop()
        if self.minStack[-1] == tmp:
            self.minStack.pop()

    def top(self) -> int:
        return self.stack[-1]

    def min(self) -> int:
        return self.minStack[-1]


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

执行结果:

image.png