NC90.包含min函数的栈

一、题目描述

定义栈的数据结构，请在该类型中实现一个能够得到栈中所含最小元素的 min 函数，输入操作时保证 pop、top 和 min 函数操作时，栈中一定有元素。

此栈包含的方法有：
push(value):将value压入栈中
pop():弹出栈顶元素
top():获取栈顶元素
min():获取栈中最小元素

二、示例

示例一：
输入: [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
输出: -1,2,1,-1
解析:
"PSH-1"表示将-1压入栈中，栈中元素为-1
"PSH2"表示将2压入栈中，栈中元素为2，-1
“MIN”表示获取此时栈中最小元素 --> 返回-1
"TOP"表示获取栈顶元素–>返回2
"POP"表示弹出栈顶元素，弹出2，栈中元素为-1
"PSH1"表示将1压入栈中，栈中元素为1，-1
"TOP"表示获取栈顶元素–>返回1
“MIN”表示获取此时栈中最小元素–>返回-1

示例二：
输入： [“PSH-1”,“PSH2”,“MIN”,“TOP”,“POP”,“PSH1”,“TOP”,“MIN”]
输出：-1,2,1,-1

三、主要思想

这道题的思想是借助辅助栈实现找到最小值，这样做的时间复杂度是O(1)，我们需要一个dataStack栈用来存放数据，一个minStack栈用来存放当前dataStack内的最小值。

假设我们要入栈的序列是1、2、3、0、4，我们设计这样一个结构，当每一次插入数据道dataStack之前，都将要插入的数据与minStack的栈顶元素进行比较。

如果要插入的数据比minStack的栈顶元素小，则将这个要插入的数据同时插入到dataStack和minStack中，比如下图插入数据0的时候。

如果要插入的数据不比minStack的栈顶元素小，则将这个要插入的数据插入到dataStack中，再取minStack中的栈顶元素继续插入的minStack中。

这样做的好处就是，minStack的栈顶元素一定是当前dataStack里的最小值，当dataStack要弹出数据的时候，minStack同样也弹出数据，这样两个栈同步弹出，就能保证minStack的栈顶元素始终都是dataStack里的最小值。

比如dataStack要弹出栈顶元素4时，minStack也弹出栈顶元素0，dataStack再弹出栈顶元素0时，minStack也弹出栈顶元素0，之后dataStack里的最小值就是1了，而minStack的栈顶元素此时也是1。

class Solution {
private:
    stack<int> dataStack;// 存放数据栈
    stack<int> minStack;// 存放最小值栈
public:
    void push(int value) {
        // 如果minStack为空，直接插入
        if(minStack.empty())
        {
            minStack.push(value);
            dataStack.push(value);
        }
        else
        {
            // 如果value比minStack栈顶元素小，则插入minStack
            if(minStack.top() > value)
            {
                minStack.push(value);
                dataStack.push(value);
            }
            // 否则只插入dataStack,minStack插入栈顶的值
            else
            {
                dataStack.push(value);
                minStack.push(minStack.top());
            } 
        }

    }
    void pop() {
        dataStack.pop();
        minStack.pop();
    }
    int top() {
        return dataStack.top();
    }
    int min() {
        return minStack.top();
    }
};