首先我们可以想到一种很简单的方法:将元素压入栈中,当第二个元素压栈的时候,与第一个元素进行比较,然后使较大的元素出栈,压栈结束后,栈中存留的便是最小的元素。但是,如果我们要想得到下一个最小的元素呢?
另一种方法:便是建立一个辅助栈来存放最小值。
例:首先在空的数据栈中压入数字3,3是此时的最小值,把3也同时压入辅助栈;
接下来向数据栈中压入数字4,由于4>3,因此仍然向辅助栈里压入数字3;
继续向数据栈中压入2,由于2<3,因此把最小值更新为2,并把2压入辅助栈;
即,遇到一个值就与之前更新的最小值进行比较,小于最小值就压入辅助栈,如果比当前最小值大,就将当前最小值压入辅助栈。
package Offer;
/**
* Created by zhangshuyou on 2015/5/19.
*/
import java.util.Stack;
/**
* 包含min函数的栈:
* 定义栈的数据结构,请在该类型中实现一个能够得到栈最小元素的min函数
*/
public class push {
Stack<Integer> dataStack = new Stack<Integer>(); //数据栈
Stack<Integer> minStack =new Stack<Integer>(); //最小栈:存放最小值
public void push(int node) {
dataStack.push(new Integer(node));
if(minStack.isEmpty() || node < minStack.peek().intValue()) //栈判空不可用minStack == null
minStack.push(new Integer(node));
else
minStack.push(minStack.peek());
}
public void pop() {
if(dataStack.isEmpty() || minStack.isEmpty())
return;
dataStack.pop();
minStack.pop();
}
public int top() {
if(minStack.isEmpty() || dataStack.isEmpty())
return 0;
return minStack.peek();
}
public int min() {
return top();
}
}