每天一道算法题——包含min函数的栈

前言
这段时间到了公司，各种事情有点多。好几天没有坚持做算法题了。今天重新捡起来。每天半个小时，剑指offer！
题目描述
定义栈的数据结构，请在该类型中实现一个能够得到栈最小元素的min函数。
测试用例:
[“PSH3”,”MIN”,”PSH4”,”MIN”,”PSH2”,”MIN”,”PSH3”,”MIN”,”POP”,”MIN”,”POP”,”MIN”,”POP”,”MIN”,”PSH0”,”MIN”]
对应输出应该为:
3,3,2,2,2,3,3,0

分析：
这个问题的重点在于怎么去记住栈中最小的那个值，而且在删去这个最小值的时候，最小值能够得到更新。这里采用创建一个辅助栈的方法帮助记忆栈中的最小值。当获得最小值时，置于栈顶，在push和最小值等值的值时，也加入辅助栈。在pop时和辅助栈顶的值进行比较，如果相等就也进行pop操作，否则就保持原样。调用min方法的时候，就返回辅助栈栈顶的值即可。

源码：

import java.util.Stack;

public class Test1 {
    public class Solution {
        Integer temp = null;
        Stack<Integer> min = new Stack<Integer>();//此为辅助栈
        Stack<Integer> data = new Stack<Integer>();//此为主栈
        public void push(int node) {
            if (temp!=null) {//如果栈中已经有数据
                if(node <= temp){//和之前的最小值进行比较
                    min.push(node);//若node不大于之前的最小值，就将之push到min栈中
                    temp = node;
                }
                data.push(node);//进行主栈基础的数据push
            }else {//若栈中没有数据
                temp = node;
                data.push(node);
                min.push(node);
            }
        }       
        public void pop() {
            int num = min.pop();
            if(data.pop()!=num){min.push(num);}//此处主栈已经进行了pop操作
        }

        public int top() {
            int num = data.peek();
            return num;
        }

        public int min() {
            int num = min.peek();//获取min栈栈顶元素即可
            return num;
        }
    }
}

运行测试：

第一种：
运行时间：15ms
占用内存：8628k

总结：
要抓住题目的核心，使用任何数据结构进行辅助的操作都可以。另外也可使用ArrayList来做辅助存储的容器。但是借助栈来实现总是要方便一些的。可以少处理很多不必要的逻辑。