包含min函数的栈

包含min函数的栈

题目描述

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

思路, 若使用标记用来标记最小元素的位置, 若弹出的是最小元素, 还需要次最小元素的标记位, 若再弹出的是次最小元素, 又需要次次最小元素, 依次类推, 则不能确定需要多少个标记位

使用两个栈, 一个用来当数据栈, 一个用来装最小值的栈, 两个栈的压入与弹出操作同时进行, 栈的最小值永远位于最小值栈的栈顶

当压入元素小于最小值栈时, 数据栈与最小值栈同时压入相同元素

当压入元素大于等于的于最小值栈的栈顶时, 数据栈正常压入, 最小值栈压入最小值栈的栈顶元素

这样, 两个栈弹出操作同步执行, 当弹出元素不是最小值时, 由于之前压入该值时, 最小栈压入是当时栈顶元素, 因此, 弹出后最小栈的栈顶还是最小值

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