题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数,在该栈中,调用min、push及pop的时间复杂度都是O(1)。
思路:建立一个新的栈minStack,每次push的时候,和minStack中的栈顶元素进行比较,如果比栈顶元素小,就入栈,minStack中栈顶元素一直是最小的数。如果pop,如果pop大小和minStack中的栈顶元素大小相同,就也将minStack中的栈顶元素出栈。
在这里我没有使用《剑指Offer》中每次入栈都将当前最小值入栈,我采用的方法是:如果当前要入栈的值小于等于当前最小值,就在minStack中入栈,出栈时,如果出栈的数等于minStack的栈顶元素,就将minStack的栈顶元素出栈,因为之前入栈时是小于等于入栈,所以即使栈内还有同样大小的数,也不会出错,采用这样的方法主要是为了节省栈的空间,而且不必每次出栈时minStack也要出栈,只需要判断就行了。
代码:
stack<int> data, minStack;
void push(int value)
{
data.push(value);
if(minStack.empty())
{
minStack.push(value);
}
else
{
if(value <= minStack.top())
{
minStack.push(value);
}
}
}
void pop()
{
if(data.top() == minStack.top())
{
minStack.pop();
}
data.pop();
}
int top()
{
return data.top();
}
int min()
{
return minStack.top();
}