问题:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
注意:保证测试中不会当栈为空的时候,对栈调用pop()或者min()或者top()方法。
输入:类
输出:类中的成员函数实现
思路:设计一个辅助栈,入栈时,将数与辅助栈的栈顶数字相比较,如果数大于栈顶数,则保存栈顶数字到辅助栈。否则,将数保存于辅助栈,同时更新最小值。
代码:
class Solution {
public:
stack<int> minStack;
stack<int> s;
void push(int value) {
if(minStack.empty())
{
minStack.push(value);
s.push(value);
}
else if(value>minStack.top())
{
minStack.push(minStack.top());
s.push(value);
}
else
{
minStack.push(value);
s.push(value);
}
}
void pop() {
if(!minStack.empty())
{
minStack.pop();
s.pop();
}
else
{
throw;
}
}
int top() {
if(!minStack.empty())
return minStack.top();
else
throw;
}
int min() {
if(!minStack.empty())
return minStack.top();
else
throw;
}
};
复杂度分析:时间复杂度为O(1).空间复杂度为O(n).