题目:定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min,push及pop的时间复杂度都是O(1)
这道题第一反应就是用一个额外的变量存储栈中的最小元素,但是仔细一想,在取出这个最小元素之后呢?怎么得到下一个最小元素。也就是说,在压入这个最小元素之前,我们还要把次小元素保存起来。仔细分析过程,可以这样做,把每次的最小元素都保存起来,放在另外一个辅助栈里。实现代码如下:
public class Problem30 {
private ArrayList<Integer> dataList=new ArrayList<>();
private ArrayList<Integer> minList=new ArrayList<>();
private Integer minValue=Integer.MAX_VALUE;
public void push(int node){
dataList.add(node);
if(node<Integer.MAX_VALUE){
minList.add(node);
minValue=node;
}else{
minList.add(minValue);
}
}
public int getSize(){
return dataList.size();
}
//栈后进后出 出栈就是取出List最后一个元素
public void pop(){
int end=getSize()-1;
dataList.remove(end);
minList.remove(end);
minValue=minList.get(getSize()-1);
}
public int min(){
return minValue;
}
这里我们用ArrayList实现逻辑上栈的数据结构。dataList是实际的数据存储的栈,minList是保存对应当前栈中最小元素的栈。每次元素入栈,都把当前数据中的最小值入辅助栈;每次出栈,辅助栈也弹出栈顶元素。辅助栈的入栈规则,保证了下一栈顶元素也是剩下元素中的最小元素。