python 最小栈
生活随笔
收集整理的這篇文章主要介紹了
python 最小栈
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
| 最小棧
設(shè)計(jì)一個支持 push ,pop ,top 操作,并能在常數(shù)時間內(nèi)檢索到最小元素的棧。 push(x) —— 將元素 x 推入棧中。 pop() —— 刪除棧頂?shù)脑亍?top() —— 獲取棧頂元素。 getMin() —— 檢索棧中的最小元素。示例: 輸入: ["MinStack","push","push","push","getMin","pop","top","getMin"] [[],[-2],[0],[-3],[],[],[],[]]輸出: [null,null,null,null,-3,null,0,-2]解釋: MinStack minStack = new MinStack(); minStack.push(-2); minStack.push(0); minStack.push(-3); minStack.getMin(); --> 返回 -3. minStack.pop(); minStack.top(); --> 返回 0. minStack.getMin(); --> 返回 -2.| 題解
class MinStack:"""題目要求在常數(shù)時間內(nèi)獲得棧中的最小值,因此不能在 getMin() 的時候再去計(jì)算最小值,最好應(yīng)該在 push 或者 pop 的時候就已經(jīng)計(jì)算好了當(dāng)前棧中的最小值。 可以用一個棧,這個棧同時保存的是每個數(shù)字 x 進(jìn)棧的時候的值 與 插入該值后的棧內(nèi)最小值。即每次新元素 x 入棧的時候保存一個元組:(當(dāng)前值 x,棧內(nèi)最小值)。 這個元組是一個整體,同時進(jìn)棧和出棧。即棧頂同時有值和棧內(nèi)最小值,top()函數(shù)是獲取棧頂?shù)漠?dāng)前值,即棧頂元組的第一個值; getMin() 函數(shù)是獲取棧內(nèi)最小值,即棧頂元組的第二個值;pop() 函數(shù)時刪除棧頂?shù)脑M。 每次新元素入棧時,要求新的棧內(nèi)最小值:比較當(dāng)前新插入元素 x 和 當(dāng)前棧內(nèi)最小值(即棧頂元組的第二個值)的大小。 新元素入棧:當(dāng)棧為空,保存元組 (x, x);當(dāng)棧不空,保存元組 (x, min(此前棧內(nèi)最小值, x))) 出棧:刪除棧頂?shù)脑M。"""def __init__(self):self.stack = []def push(self, val: int) -> None:if not self.stack:self.stack.append((val, val))else:self.stack.append((val, min(val, self.stack[-1][-1])))def pop(self) -> None:self.stack.pop()def top(self) -> int:return self.stack[-1][0]def getMin(self) -> int:return self.stack[-1][-1]# Your MinStack object will be instantiated and called as such: # obj = MinStack() # obj.push(val) # obj.pop() # param_3 = obj.top() # param_4 = obj.getMin()總結(jié)
以上是生活随笔為你收集整理的python 最小栈的全部內(nèi)容,希望文章能夠幫你解決所遇到的問題。
- 上一篇: python 有效的括号
- 下一篇: python 用一个数组实现三个栈