【剑指offer】20- 包含min函数的栈

本文系《剑指offer》的刷题记录，通过牛客网在线平台测试通过。
在线测试平台：牛客网
编程资料获取：CodeLab

1- Description

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

2-Solution

注意这里强调了得到最小元素要求时间复杂度是O(1)，所以可以考虑将最小的元素存储起来备用，这里采取的方式是创建一个辅助栈进行最小元素的存储。

下面就以连续压入3、 4、 2、 1之后接连两次弹出栈顶元素，再压入0这一过程进行简单分析。

• 首先进行入栈操作，两个栈均添加同一元素，之后只有当前元素小于等于辅助栈顶元素辅助栈才有元素入栈（比如②就没有进行入栈操作）

• 随后的出栈操作，只要辅助栈的栈顶元素小于等于当前元素则二者均应该弹出，否则只有主栈弹出

根据上述算法步骤，可以完成如下解决代码：

class Solution {
    stack<int> mdata;
    stack<int> mmin;
public:
    void push(int value) {
        mdata.push(value);
        if(mmin.size() == 0 || value <= mmin.top())  mmin.push(value);//一个存入最小值的栈
    }
    void pop() {
        if(mdata.top() == mmin.top())  mmin.pop();//只有当主栈和辅助栈栈顶元素相等时才弹出
        mdata.pop();
    }
    int top() {
        return mdata.top();
    }
    int min() {
        return mmin.top();
    }
};