剑指offer面试题[21]包含min函数的栈

来源:互联网 发布:网络大电影的受众 编辑:程序博客网 时间:2024/06/06 05:15

题目:

定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数。

分析:

       看到这个问题,我们的第一反应可能是每次压入一个新元素进栈时,将栈里所有的元素排序,让最小的元位于栈顶,这样就能在O(1)时间得到最小元素了。但这种思路不能保证最后压入栈的元素就能最先出栈,因此这个数据结构已经不是栈了。

       我们接着想在栈里添加一个成员变量存放最小的元素。每次压入一个新元素进栈的时候,如果该元素比当前最小的元素还要小,则更新最小元素。面试官听到这种思路就会问:如果最小的元素被弹出栈了,如何得到下一个最小的元素呢?

        分析到这里我们发现仅仅添加一个成员变量存放最小元素是不够的,也就是说当最小元素被弹出的时候,我们希望得到次小元素。因此在压入这个最小元素之前,我们要把次最小元素保存起来。

      是不是可以把每次的最小元素(之前的最小元素和新压入栈的元素两者的较小值)都保存起来放到另外一个辅助栈里呢?我们可以举例来分析以下把元素压入或者弹出栈的过程。

        栈内压入3、4、2、1后连续两次弹出栈顶数字再压入0时,数据栈、辅助栈和最小值的状态如下。

                                   
  代码如下:

class Solution {
public:
    stack<int> dataStack,minStack;
    void push(int value) {
        dataStack.push(value);
        if(minStack.empty()||value<minStack.top())
            minStack.push(value);
        else
            minStack.push(minStack.top());         
    }
    
    void pop() {
         dataStack.pop();
         minStack.pop();
    }
    
    int top() {
         return dataStack.top();
    }
    
    int min() {
        return minStack.top();
    }
};

原创粉丝点击