Min Stack
原题链接Min Stack
实现一个栈,除了基本功能外需要提供一个接口用于返回栈中元素的最小值
基本功能使用std::stack就够了,如果每次需要最小值时都遍历栈中元素的话会很麻烦,因为栈不支持遍历(本质上是没有提供迭代器),所以只能在push的时候记录最小值
思路是这样的,使用另一个std::stack记录最小值。简称最小栈
push时记录最小值的方法是
- 如果要添加的值小于最小栈栈顶,则将其push到最小栈中
- 如果要添加的值大于最小栈栈顶,则将栈顶元素再push一遍到最小栈中
pop函数的做法是
- 弹出数值栈栈顶
- 弹出最小栈栈顶
这种做法和单调队列类似,假设当前最小栈栈顶为a,要push的元素是b
如果a > b,则将b添加到最小栈中,最小值为b,很正常
如果a < b,则再push一个a到最小栈中,最小值仍为a,满足要求。这么做的原因是在b存在期间,最小值肯定是a,所以最小栈中不需要存b。而b销毁时,从最小栈中弹出多push的a,恢复到添加b之前的状态。
代码如下
class MinStack {
public:
/** initialize your data structure here. */
MinStack() {
}
void push(int x) {
stack_.push(x);
if(min_.empty() || min_.top() > x)
min_.push(x);
else
min_.push(min_.top());
}
void pop() {
stack_.pop();
min_.pop();
}
int top() {
return stack_.top();
}
int getMin() {
return min_.top();
}
private:
std::stack<int> stack_;
std::stack<int> min_;
};
/**
* Your MinStack object will be instantiated and called as such:
* MinStack obj = new MinStack();
* obj.push(x);
* obj.pop();
* int param_3 = obj.top();
* int param_4 = obj.getMin();
*/