包含min函数的栈Java

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

思路：使用两个栈，当向栈中添加元素时，与辅助栈中元素比较，如果小，向辅助栈中添加该元素，向辅助栈中再次添加辅助栈栈顶元素。

package offer;

import java.util.Stack;

public class MinStack<T extends Comparable<? super T>> {
	
	private Stack<T> stack1;
	/*
	 * 辅助栈，存放最小值
	 */
	private Stack<T> stackMin;
	public MinStack(){
		stack1 = new Stack<T>();
		stackMin = new Stack<T>();
	}
	public void push(T t){
		stack1.push(t);
		if(stackMin.size()==0 || t.compareTo(stackMin.peek())<0)
			stackMin.push(t);
		else
			stackMin.push(stackMin.peek());
	}
	public T pop(){
		if(stack1.size()>0 && stackMin.size()>0){
			stackMin.pop();
			return stack1.pop();
		}
		return null;
	}
	public T min(){
		if(stack1.size()>0 && stackMin.size()>0)
			return stackMin.peek();
		return null;
	}
	public static void main(String[] args){
		MinStack<Integer> minStack = new MinStack<Integer>();
		minStack.push(4);
		minStack.push(3);
		minStack.push(7);
		minStack.pop();
		System.out.println(minStack.min());
	}

}