题目描述:定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
首先 栈的特征 先进后出。
因此我们只要定义一个min_stack的栈,每push一个元素都与min_stack.top()元素进行大小比较,如果比min_stack.top()小的话就push进min_stack,这样就可以保证min_stack这个栈的top元素始终为最小的元素。
class Solution
{
private:
stack<int> min_stack;
stack<int> my_stack;
public:
void push(int value)
{
my_stack.push(value);
if (min_stack.size() == 0)
{
min_stack.push(value);
}
if (min_stack.top() > value)
{
min_stack.push(value);
}
}
void pop()
{
if (my_stack.top() == min_stack.top())
{
my_stack.pop();
min_stack.pop();
}
else
{
my_stack.pop();
}
}
int top()
{
return my_stack.top();
}
int min()
{
return min_stack.top();
}
};