【簡單算法】39.最小棧
阿新 • • 發佈:2018-05-07
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.最小棧