定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。
得到一个栈中的最小值,我们需要另一个栈作为辅助,当栈为空时,压入的第一个数字则为最小值,同时将该数字压入辅助栈
来第二个数字时,与辅助栈栈顶比较,若小于栈顶则压入辅助栈作为新的最小值,若大于栈顶则再次压入栈顶值,表示此时最小值仍然为之前的值
代码如下
class Solution {
stack<int> data;
stack<int> mins; //定义辅助栈
int temp;
public:
void push(int value) {
if(mins.empty()) mins.push(value); //如果辅助栈为空,则直接压入value
else{
if(mins.top()>value) mins.push(value); //如果新值小于栈顶值,则压入value更新最小值
else mins.push(mins.top()); //否则,再次压入栈顶值,最小值不变
}
}
void pop() {
mins.pop(); //做如上push操作后,pop,top完全与主栈一致
}
int top() {
return mins.top();
}
int min() {
return mins.top(); //无论如何,栈顶值永远是当前栈中元素的最小值
}
};