一个栈是没有办法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();
}
};
京公网安备 11010502036488号