题目在这里: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(); } };