Java解决剑指offer 包含min函数的栈

题目描述

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

题目分析

看到题目，第一反应就是使用一个整数来保存栈中的最小元素，这样可以时间入栈(push)函数、和min函数的时间复杂度为O(1)，但却不能保证出栈(pop)函数的时间复杂度，因为当最小元素出栈后，我们不得不遍历栈来寻找次小值元素(即当前新的最小值元素)，因此这不是我们想要的解法。
我们不妨换一种思路，既然最小元素出栈后，我们需要重新获取次小值元素，而这步操作导致我们时间复杂度不符合要求，那我们何不把次小值元素也保存起来？
首先，我们需要一个数据栈和辅助栈。先往空的数据栈里压入数字3，此时最小值是3，因此将3压入辅助栈。接下来向数据栈压入4，此时最小值为3，因此向辅助栈压入3。第三步继续向数据栈压入2，此时最小值为2，因此向辅助栈压入2。
由于我们每次向辅助栈压入的都是压入时的最小值，因此min函数中只需要返回辅助栈的栈顶元素即可。对于pop函数，每次对数据栈的pop操作，我们需要同时对辅助栈也进行以此pop操作，当数据栈pop出的元素是最小值时，辅助栈也对应将栈顶最小值pop出栈，剩下的即为次小值；当数据栈pop出的元素不是最小值时，辅助栈在pop后，栈顶元素也仍为最小值。因此这是一个正确的解法。
代码如下。

Java代码

import java.util.Stack;

public class Solution {

    private Stack<Integer> stack = new Stack<Integer>();
    private Stack<Integer> minStack = new Stack<Integer>();
    public void push(int node) {
        stack.push(node);
        if(minStack.empty()){
            minStack.push(node);
        }else{
            int min = minStack.peek()>node?node:minStack.peek();
            minStack.push(min);
        }
    }
    
    public void pop() {
        stack.pop();
        minStack.pop();
    }
    
    public int top() {
        return stack.peek();
    }
    
    public int min() {
        return minStack.peek();
    }
}