剑指Offer系列-面试题21:包含min函数的栈

题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数,在该栈中,调用min、push及pop的时间复杂度都是O(1)。

思路:建立一个新的栈minStack,每次push的时候,和minStack中的栈顶元素进行比较,如果比栈顶元素小,就入栈,minStack中栈顶元素一直是最小的数。如果pop,如果pop大小和minStack中的栈顶元素大小相同,就也将minStack中的栈顶元素出栈。

在这里我没有使用《剑指Offer》中每次入栈都将当前最小值入栈,我采用的方法是:如果当前要入栈的值小于等于当前最小值,就在minStack中入栈,出栈时,如果出栈的数等于minStack的栈顶元素,就将minStack的栈顶元素出栈,因为之前入栈时是小于等于入栈,所以即使栈内还有同样大小的数,也不会出错,采用这样的方法主要是为了节省栈的空间,而且不必每次出栈时minStack也要出栈,只需要判断就行了。

代码:

stack 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();
}


你可能感兴趣的:(剑指Offer,剑指Offer)