LeetCode——第12题：带最小值的栈（min栈）

带最小值操作的栈（min栈）

---

题目：实现带有最小值min方法的栈
说明:
实现一个带有取最小值min方法的栈，min方法将返回当前栈中的最小值。
你实现的栈将支持push，pop 和 min 操作，所有操作要求都在O(1)时间内完成。

输入：push(1)，pop()，push(2)，push(3)，min()， push(1)，min()

输出：1，2，1

这道题总共有三种解法，我们一一说明，最后会给出最好的一种解法的代码。

第一种解法，思路：

1、新建一个辅助栈来存储输入元素的最小值；
2、如果下一个元素小于当前最小值，那么存入辅助栈中；
3、如果下一个元素大于当前最小值，则不做任何操作；
4、出栈时，如果数据栈的栈顶与辅助栈的栈顶元素相同时，辅助栈才出栈。
但是这样会存在一个问题：
如果有多个相同的元素，比如 3, 2, 2, 1, 1
那么在遇到第一个1的时候，辅助栈的最小值1已经出栈，对于第二个1，辅助栈的最小值是2，这是错误的。

第二种解法，思路：

1、还是用辅助栈，但是如果下一个元素小于当前最小值，将下一个元素存入辅助栈中；
2、当下一个元素不小于当前最小值，则将当前最小值（栈顶元素）存入辅助栈中；
3、在出栈的时候，数据栈和辅助栈都一起出栈。
不足：这样的话，辅助栈元素会增加。

第三种解法：

转载自：https://www.cnblogs.com/cane/p/3793510.html

重点：不需要额外的空间。

具体思路：栈顶保存的是最小元素，每次push比较，如果当前元素更小的话就更新栈顶，另外栈顶以下元素保存的是元素push的时候与当时的最小值的差值；pop的时候，就是看栈顶元素后的下一个元素，如果小于零，说明当前需要pop最小元素，同时更新最小元素，如果大于零，直接push（最小元素+差值）即可。

举例说明：
3 5 2 1
push:
3
2 3
2 -1 2
2 -1 -1 1
pop:
2 -1 2 输出1
2 3 输出2
3 输出5
输出3

代码如下：

/*
题目：min栈的实现，C++
*/
class MinStack {
public:
    stack<int> stk;
    MinStack() {
        // do intialization if necessary
    }

    /*
     * @param number: An integer
     * @return: nothing
     */
    void push(int number) {
        // write your code here
        if (stk.empty()) {
            stk.push(number);
        }
        else if (number < stk.top()) {
            int temp = stk.top();
            stk.pop();
            stk.push(number - temp);
            stk.push(number);
        }
        else if (number >= stk.top()) {
            int temp = stk.top();
            stk.pop();
            stk.push(number - temp);
            stk.push(temp);
        }
    }

    /*
     * @return: An integer
     */
    int pop() {
        // write your code here
        if (stk.empty()) {
            return -1;
        }
        else {
            int temp = stk.top();
            stk.pop();
            int num = temp;
            if (!stk.empty()) {
                num = stk.top();
                stk.pop();
                if (num <= 0) {
                    stk.push(temp - num);
                    num = temp;
                }
                else {
                    stk.push(temp);
                    num = temp + num;
                }
            }
            return num;

        }
    }

    /*
     * @return: An integer
     */
    int min() {
        // write your code here
        return stk.top();
    }
};