1. 程式人生 > >【簡單算法】39.最小棧

【簡單算法】39.最小棧

spa tor min HERE data 示例 ctu 最小 nbsp

題目:

設計一個支持 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.

解題思路:

使用兩個棧,一個棧保存所有的操作數,另外一個棧保存當前的最小元素。每次新元素入棧時,同時將當前棧的最小元素也入棧,出棧時同樣執行相同的操作。

   vector<int> val;/*保存所有元素×/
   vector<int> minVal;/×保存當前棧的最小元素×/

代碼實現:

class MinStack {
public:
    /** initialize your data structure here. */
    MinStack() {
        
    }
    
    void push(int x) {
        
if(val.size() > 0){ int next = min(x,minVal[minVal.size()-1]); val.push_back(x); minVal.push_back(next); }else{ val.push_back(x); minVal.push_back(x); } } void pop() { val.pop_back(); minVal.pop_back(); }
int top() { return val[val.size()-1]; } int getMin() { return minVal[minVal.size()-1]; } private: vector<int> val; vector<int> minVal; }; /** * Your MinStack object will be instantiated and called as such: * MinStack obj = new MinStack(); * obj.push(x); * obj.pop(); * int param_3 = obj.top(); * int param_4 = obj.getMin(); */

【簡單算法】39.最小棧