【牛客题霸题解】包含min函数的栈(待更新java、py)
包含min函数的栈
http://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49
一个栈是没有办法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();
}
};牛客题霸题解 文章被收录于专栏
QAQ