题解 | #包含min函数的栈#
包含min函数的栈
http://www.nowcoder.com/practice/4c776177d2c04c2494f2555c9fcc1e49
解题思路
使用一个辅助栈存储最小值,min函数直接返回辅助栈顶就可以。
代码实现
class Solution {
public:
stack<int> normal, minval;
void push(int value) {
normal.push(value);
if (minval.empty()) {
minval.push(value);
}
else {
if (value <= minval.top()) {
minval.push(value);
}
else {
minval.push(minval.top());
}
}
}
void pop() {
if(normal.empty()) return;
normal.pop();
minval.pop();
}
int top() {
if(normal.empty()) return -1;
return normal.top();
}
int min() {
return minval.top();
}
};
注意
不能想着辅助minval栈只入最小值,比如minval栈顶是1,来个元素6就不入栈了,这样的话在normal栈出栈的时候两者就不能保持同步了,结果自然会出错。