牛客剑指Offer面试题30:包含min函数的栈

题目描述

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

思路

设定一个数据栈（真实数据入栈）和一个辅助栈（每次新元素入数据栈时，将此时数据站的最小元素压入辅助栈），数据栈弹出元素时辅助栈同时也弹出栈顶元素．　任意时刻辅助栈栈顶元素即为此时数据栈中的最小元素

代码

#include <assert.h>
class Solution {
public:
    /*
    void push(int value) {
        //元素入数据栈
        m_data.push(value);
        //如辅助栈为空　或　辅助栈顶元素＞欲入栈元素
        if(m_min.size() == 0 || m_min.top() > value)
            m_min.push(value);
        //否则，将辅助栈顶元素再次入栈
        else
            m_min.push(m_min.top());
    }
    void pop() {
        //两栈均不空时，分别pop
        assert(m_data.size() > 0 && m_min.size() > 0);
        m_data.pop();
        m_min.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        assert(m_data.size() > 0 && m_min.size() > 0);
        return m_min.top();
    }
    */
        void push(int value) {
        //元素入数据栈
        m_data.push(value);
        //如辅助栈为空　或　辅助栈顶元素＞欲入栈元素
        if(m_min.size() == 0 || m_min.top() > value)
            m_min.push(value);
        //否则，将辅助栈顶元素再次入栈
        else
            m_min.push(m_min.top());
    }
    void pop() {
        //两栈均不空时，分别pop
        //assert(m_data.size() > 0 && m_min.size() > 0);
        m_data.pop();
        m_min.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        //assert(m_data.size() > 0 && m_min.size() > 0);
        return m_min.top();
    }
private:
    //数据栈
    stack<int> m_data;
    //辅助栈
    stack<int> m_min;
    
};