剑指offer-栈-包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。

思路

• 第一想法
  类似于找到数组中的最小元素（设置一个min变量暂存当前最小值），设置一个存储变量。但是问题随之出现，如果最小元素出栈，那么之后再调用min( )方法将没有最小值的记录。
• 过渡想法
  尝试用数组存储？？这样可以保存多个值。但是数组的空间大小不能确定，所以考虑动态数组ArrayList？？
• 最终想法
  既然都想到动态数组了，莫不如使用一个辅助栈（使用相同数据结构的辅助空间），存储当前的最小值序列。

• 详细分析（node表示元素节点，s表示原栈，assist表示辅助栈）

  push() ⇒ node一定在s入栈；当assist为空时，node在assist入栈；当node < assist栈顶元素时，即出现新的最小元素，node在assist入栈。

  pop() ⇒ s正常出栈；当s出栈元素等于assist栈顶元素时，assist栈顶出栈。

  top() ⇒ 获取s栈顶值，不出栈。

  min() ⇒ 获取当前assist栈顶元素

MyCode

public class MinStack {

    //assist min stack
    Stack<Integer> minStack = new Stack<>();
    Stack<Integer> s = new Stack<>();

    public void push(int node) {
        if(minStack.isEmpty())
            minStack.push(node);
        if(node < minStack.peek()){
            s.push(node);
            minStack.push(node);
        }else{
            s.push(node);
        }
    }

    public void pop() {
        if (!s.isEmpty() && !minStack.isEmpty()) {
            if (s.peek() == minStack.peek())
                minStack.pop();
            s.pop();
        }
    }

    public int top() {
        return s.peek();
    }

    public int min() {
        if (minStack.isEmpty())
            System.exit(0);
        return minStack.peek();
    }
}

总结

• 边缘判定：出栈确保栈不空
• 针对出现问题，分析问题产生原因。本题中，最小值持续获取决定了需要辅助空间；辅助空间如何选取，选取相同的数据结构。