题目
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
代码
另外设置一个栈存放最小元素,从栈顶到栈底从小到大排列。
class Solution {
public:
void push(int value) {
s.push(value);
if(smin.empty())
smin.push(value);
else if(smin.top()>value)
smin.push(value);
}
void pop() {
if(s.top()==smin.top())
smin.pop();
s.pop();
}
int top() {
return s.top();
}
int min() {
return smin.top();
}
private:
stack<int> s;
stack<int> smin;
};