java之包含min函数的栈

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

package Stackqueue;

import java.util.Stack;

public class StackMinMain {
	Stack<Integer> m_data = new Stack<Integer>();
	Stack<Integer> m_min = new Stack<Integer>();

	public void push(int value) {
		m_data.push(value);
		if (m_min.size() == 0 || value < m_min.firstElement())
			m_min.push(value);
		else
			m_min.push(m_min.firstElement());
	}

	public void pop() {
		assert (m_data.size() > 0 && m_min.size() > 0);
		m_data.pop();
		m_min.pop();

	}

	public int min() {
		assert (m_data.size() > 0 && m_min.size() > 0);
		return m_min.peek();
	}

	public static void main(String[] args) {
		StackMinMain stackMinMain = new StackMinMain();
		int[] a = { 3, 4, 2, 1 };
		for (int i = 0; i < a.length; i++)
			stackMinMain.push(a[i]);
		System.out.println(stackMinMain.min());

	}

}

结果显示：

1