剑指offer-面试题21:包含min函数的栈

题目

定义栈的数据结构，请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中，调用min，push及pop的时间复杂度都是O(1)。

解

import java.util.Stack;

public class StackWithMin {

    Stack<Integer> m_data=new Stack<Integer>();//数据栈
    Stack<Integer> m_min=new Stack<Integer>();//辅助栈，用来存储当前数据集对应的最小元素

    void  push(Integer value){
        m_data.push(value);

        if(m_min.size()==0||value<m_min.peek()){
            m_min.push(value);
        }else{
            m_min.push(m_min.peek());
        }
    }

    void pop(){
        if(m_min.size()>0 && m_data.size()>0){
            m_data.pop();
            m_min.pop();
        }
    }

    Integer min(){
        if(m_min.size()>0 && m_data.size()>0){
            return m_min.peek();
        }else{
            return null;
        }
    }


    public static void main(String[] args) {
        // TODO Auto-generated method stub

    }

}