leetcode 155. 最小棧 (Easy)
阿新 • • 發佈:2018-12-09
題目:
設計一個支援 push,pop,top 操作,並能在常數時間內檢索到最小元素的棧。
- push(x) -- 將元素 x 推入棧中。
- pop() -- 刪除棧頂的元素。
- top() -- 獲取棧頂元素。
- getMin() -- 檢索棧中的最小元素。
示例:
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(object): def __init__(self): """ 初始化棧,最小值 initialize your data structure here. """ self.stack = [] self.min = None def push(self, x): """ 入棧 :type x: int :rtype: void """ if self.min == None or self.min > x: self.min = x self.stack.append(x) def pop(self): """ 刪除棧頂元素 :rtype: void """ stackTop = self.stack.pop() if len(self.stack) == 0: self.min = None if stackTop == self.min: self.min = self.stack[0] for i in self.stack: if i < self.min: self.min = i def top(self): """ 獲取棧頂元素 :rtype: int """ if self.stack ==None: return 'error' else: return self.stack[-1] def getMin(self): """ 檢索棧中最小元素 :rtype: int """ return self.min # Your MinStack object will be instantiated and called as such: # obj = MinStack() # obj.push(x) # obj.pop() # param_3 = obj.top() # param_4 = obj.getMin()
思路:設定一個最小值變數,在進棧出棧的時候都計算出棧中最小的值。這樣在最小值計算函式裡面直接呼叫最小值變數的值即可。