包含min函数的栈

包含min函数的栈

1.题目描述

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

2.思路

设置一个栈m_data，一个辅助栈m_min，压栈时压入m_data，如果待压入元素比m_min的栈顶元素小，则也把该元素压入m_min，m_min记录了当前的最小元素。pop栈时，如果m_data的栈顶元素和m_min的栈顶元素相等时，则同时弹出m_data和m_min的栈顶元素。取top时直接取m_data的top()，取min时，直接取m_min的top()。

3.代码

class Solution {
public:
    void push(int value) {
        m_data.push(value);
        if(m_min.empty()){
            m_min.push(value);
        }
        else if(m_min.top() > value){
            m_min.push(value);
        }
    }
    void pop() {
        if(m_data.top() == m_min.top()){
            m_min.pop();
        }
        m_data.pop();
    }
    int top() {
        return m_data.top();
    }
    int min() {
        return m_min.top();
    }
private:
    stack<int> m_data;
    stack<int> m_min;
};

4.复杂度分析

时间复杂度：O(1)
空间复杂度：O(n)

来源：CSDN

作者：overlordmax

链接：https://blog.csdn.net/jiangdongxiaobawang/article/details/104044433