LeetCodeEasy-【面試題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 次

注意:本題與主站 155 題相同:https://leetcode-cn.com/problems/min-stack/

來源:力扣(LeetCode)
鏈接:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof
著作權歸領釦網絡所有。商業轉載請聯繫官方授權,非商業轉載請註明出處。

思路:輔助棧

用一個data棧存數據,利用一個輔助棧help來記入當前位置(即[0, 當前位置])的最小值。這個只需要在data壓棧是判斷即可。
在這裏插入圖片描述
在這裏插入圖片描述

class MinStack:

    def __init__(self):
        """
        initialize your data structure here.
        """
        self.data = [] # 數據棧 
        self.help = [] # 輔助棧

    def push(self, x: int) -> None:
        self.data.append(x)
        if self.help == [] or x <= self.help[-1]:
            self.help.append(x)

    def pop(self) -> None:
        if self.data.pop() == self.help[-1]:
            self.help.pop()

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

    def min(self) -> int:
        return self.help[-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()
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章