一个栈是没有办法O(1)弹出最小值的,所以需要设置一个辅助栈来随时取出目前的最小值。
GetMin(): 辅助栈的栈顶元素表示当前的最小值
push(int x): 当插入一个元素的时候,和辅助栈的栈顶元素比较,如果插入的元素<=当前栈顶元素的话,就把这个元素也插入到辅助栈中
pop(): 当弹出一个元素的时候,如果弹出的元素==栈顶元素的话,就把辅助栈的栈顶元素也弹出
c++
class Solution { public: stack<int> S; stack<int> minS; void push(int value) { S.push(value); if(minS.empty()||value <= minS.top()) { minS.push(value); } } void pop() { int t= S.top(); S.pop(); if(t == minS.top()) { minS.pop(); } } int top() { return S.top(); } int min() { return minS.top(); } };