剑指offer 21:包含min函数的栈

题目描述

定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

解题思路

要求在O(1)时间内完成,由于栈后进先出的原则,不能出现破坏栈结构的事情。因而采用以空间换时间的思路,建立辅助栈,辅助栈与数据栈的出入栈操作同步进行,区别在于数据栈压入当前元素,辅助栈压入当前情况下的最小元素,于是,需要定义一个临时变量来记录上一时刻的最小元素,并与当前元素比较,记录当前情况下的最小元素。出栈时,两个栈同步操作。

 

C++代码实现:

class Solution {
public:
    stack<int> m_data,auxiliary;
    int minVal;
    void push(int value) {
        if(m_data.size()==0){
         minVal=value;   
        }else{
            minVal=minVal>value?value:minVal;
        }
        m_data.push(value);
        auxiliary.push(minVal);
    }
    void pop() {
        m_data.pop();
        auxiliary.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        return auxiliary.top();
    }
};

 

posted @ 2019-10-05 10:48  fancy_li  阅读(112)  评论(0编辑  收藏  举报