1、题目描述
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
2、解题思路
定义两个栈stack1和min_stack,stack1用来存储数据,min_stack用来记录stack1的最小值,假设入栈序列为:4、5、6、3、2、1,那么min_stack各轮次对应的min值就是:4、4、4、3、2、1。这样min_stack的栈顶的值就是stack1中数据的最小值。出栈时把两个栈中的元素同时弹出即可。
代码如下:
class Solution {
public:
void push(int value) {
stack1_.push(value);
if(min_stack_.empty())
min_stack_.push(value);
else if (min_stack_.top() < value)
min_stack_.push(min_stack_.top());
else
min_stack_.push(value);
}
void pop() {
if(!stack1_.empty() && !min_stack_.empty())
{
stack1_.pop();
min_stack_.pop();
}
}
int top() {
return stack1_.top();
}
int min() {
return min_stack_.top();
}
private:
stack stack1_;//主要存数据的栈
stack min_stack_;//存最小值的栈
};