题目在这里:https://oj.leetcode.com/problems/min-stack/
和普通的栈只有一点区别,就是要能够支持常数时间的最小值抽取。只要维护一个单调队列就可以了,从队首到队尾非降排序。
对于三种操作:
① 新来一个元素,如果该元素不大于队首的元素,则入队,否则不入队(直观上理解,你来得晚而且数值还大,以后肯定不会是最小值的);
② 删除元素时,比较当前待删除的元素和队首元素是否相等,相等则把队首元素删除,否则什么都不做;
③ 取最小值很简单啦,把队首元素作为结果返回就可以了。
class MinStack {
public:
stack<int> data;
stack<int> mn;
void push(int x) {
data.push(x);
if (mn.empty() || mn.top() >= x) mn.push(x);
}
void pop() {
if (data.top() == mn.top()) mn.pop();
data.pop();
}
int top() {
return data.top();
}
int getMin() {
return mn.top();
}
};