【剑指offer】面试题30：包含 min 函数的栈

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

 我们可以使用两个栈，一个数据栈，另一个为辅助栈。辅助栈用于存放最小元素和每次压栈时的次小元素（之前的最小元素和新压入栈的元素两者之间的最小值或相等）。

如果每次都把最小元素压入辅助栈，那么就能保证辅助栈顶一直都是最小元素。当最小元素从数据栈中被弹出后，同时弹出辅助栈的栈顶元素，此时辅助栈的新栈顶元素就是下一个最小元素。

下面看个案例再了解下这个过程：

步骤1：首先往空的数据栈中压入数字3，显然现在3是最小值，则也在辅助栈中压入3；

步骤2：往数据栈中压入数字4，由于4大于之前的最小值，因此仍然往辅助栈中压入3；

步骤3：往数据栈中压入数字2，由于2小于之前的最小值3，因此把最小值更新为2，并把2压入辅助栈；

步骤4：往数据栈中压入数字1，由于1小于之前的最小值2，因此把最小值更新为1，并把1压入辅助栈；

步骤5：弹出数据栈中的1，辅助栈的栈顶元素1也随之弹出，则此时辅助栈的栈顶元素为2；

步骤6：弹出数据栈中的2，辅助栈的栈顶元素2也随之弹出，则此时辅助栈的栈顶元素为3；

步骤7：往数据栈中压入数字0，由于0小于之前的最小值3，因此把最小值更新为0，并把0压入辅助栈。

..........

步骤	操作	数据栈	辅助栈	最小值
1	压入3	3	3	3
2	压入4	3，4	3，3	3
3	压入2	3，4，2	3，3，2	2
4	压入1	3，4，2，1	3，3，2，1	1
5	弹出	3，4，2	3，3，2	2
6	弹出	3，4	3，3	3
7	压入0	3，4，0	3，3，0	0

将上面的过程翻译成代码即为：

public class MinStack {

	Stack<Integer> dataStack = new Stack<Integer>();  // 数据栈
	Stack<Integer> minStack = new Stack<Integer>();   // 辅助栈
	
	public void push(int node){
		dataStack.push(node);     // 压进数据栈
		if(minStack.isEmpty() || node < minStack.peek()){
			minStack.push(node);  // 压进辅助栈
		}else{
			// 如果要压进数据栈的元素比当前最小值大，则将最小值再压入辅助栈一次
			minStack.push(minStack.peek());
		}
	}
	
	public void pop(){
		dataStack.pop();
		minStack.pop();
	}
	
	public int top(){
		return dataStack.peek();
	}
	
	public int min(){
		return minStack.peek();
	}
}