LeetCode 155. 最小栈

题目链接：https://leetcode-cn.com/problems/min-stack/

1.一个链表

思路如下：

Java代码如下：

class MinStack {
    private ListNode head;

    public MinStack() {
        head = new ListNode(-1, Integer.MAX_VALUE, null);
    }
    
    public void push(int val) {
        head = new ListNode(val, Math.min(val, head.min), head);
    }
    
    public void pop() {
        head = head.next;
    }
    
    public int top() {
        return head.val;
    }
    
    public int getMin() {
        return head.min;
    }

    private static class ListNode {
        public int val;
        public int min;
        public ListNode next;
        public ListNode(int val, int min, ListNode next) {
            this.val = val;
            this.min = min;
            this.next = next;
        }
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(val);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */

2.两个栈

压入

假设当前数据为 val，先直接将 val 压入 stackData；

如果 stackMin 为空，则直接将 val 压入 stackMin；

如果 stackMin 不为空，则将 val 和 stackMin 栈顶元素两者中的较小者压入 stackMin。

弹出

同时弹出 stackData 和 stackMin 的栈顶元素。

查询最小值

stackMin 的栈顶元素始终是当前 stackData 中的最小值。

Java代码如下：

class MinStack {
    private Stack<Integer> stackData;
    private Stack<Integer> stackMin;

    public MinStack() {
        stackData = new Stack<>();
        stackMin = new Stack<>();
    }

    public void push(int val) {
        stackData.push(val);
        if (stackMin.isEmpty()) {
            stackMin.push(val);
        } else {
            stackMin.push(Math.min(val, stackMin.peek()));
        }
    }
    
    public void pop() {
        stackData.pop();
        stackMin.pop();
    }
    
    public int top() {
        return stackData.peek();
    }
    
    public int getMin() {
        return stackMin.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(val);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */

C++代码如下：

class MinStack {
public:
    stack<int> stackData, stackMin;
    
    MinStack() {

    }
    
    void push(int val) {
        stackData.push(val);
        if(stackMin.empty() || val < stackMin.top())    stackMin.push(val);
        else    stackMin.push(stackMin.top());
    }
    
    void pop() {
        stackData.pop();
        stackMin.pop();
    }
    
    int top() {
        return stackData.top();
    }
    
    int getMin() {
        return stackMin.top();
    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(val);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */

在空间上，还可以进行常数级别的优化。

压入

假设当前数据为 val，先直接将 val 压入 stackData；

如果 stackMin 为空，则直接将 val 压入 stackMin；

如果 stackMin 不为空：如果 val 小于等于 stackMin 的栈顶元素，则将 val 压入 stackMin；否则，stackMin 不压入任何内容。

弹出

stackData 肯定要弹出其栈顶元素。

由于 stackMin 中存在的元素是从栈底到栈顶逐渐变小的，stackMin 栈顶元素既是 stackMin 的最小值，也是当前 stackData 的最小值。所以不会出现 stackData 栈顶元素比 stackMin 栈顶元素更小的情况，stackData 栈顶元素只可能大于或等于 stackMin 栈顶元素。

因此，当 stackData 栈顶元素等于 stackMin 栈顶元素时，stackMin 也随之弹出栈顶元素；当 stackData 栈顶元素大于 stackMin 栈顶元素 时，stackMin 不弹出栈顶元素。

查询最小值

stackMin 的栈顶元素始终是当前 stackData 中的最小值。

C++代码如下：

class MinStack {
public:
    stack<int> stackData, stackMin;

    MinStack() {

    }
    
    void push(int val) {
        stackData.push(val);
        if(stackMin.empty() || val <= stackMin.top())   stackMin.push(val);
    }
    
    void pop() {
        if(stackData.top() == stackMin.top())   stackMin.pop();
        stackData.pop();
    }
    
    int top() {
        return stackData.top();
    }
    
    int getMin() {
        return stackMin.top();
    }
};

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack* obj = new MinStack();
 * obj->push(val);
 * obj->pop();
 * int param_3 = obj->top();
 * int param_4 = obj->getMin();
 */