题目:定义栈的数据结构,在该类型中实现一个能够得到栈的最小元素的min函数,在该栈中,调用min,push,pop的时间复杂度都是O(1)
分析:
1.第一反应可能是每次压栈时,将栈所有元素排序,让最小的元素位于栈顶,这样就能够在O(1)时间得到最小元素,但是这样不能保证最后压栈的元素能先出栈,即栈的后进先出的特性不存在
2.可以在栈里面添加一个成员变量,每次压栈时,如果该元素比当前最小的元素还要小,则更新最小元素,但是要求得次小元素的话就没办法了
3.可增加一个辅助栈,用data保存原数,min保存每次入栈时当前栈中的最小数字,eg
import java.util.*;
public class wr20stackMin {
Stack<Integer> data=new Stack<>();
Stack<Integer> min=new Stack<>();
Integer temp=null;
public void push(int node){
if(temp!=null){
if(node<=temp){
temp=node;
min.push(node);
}
data.push(node);
}else{
temp=node;
data.push(node);
min.push(node);
}
}
public void pop(){
int num=data.pop();
int num2=min.pop();
if(num!=num2){
min.push(num2);
}
}
public int top(){
int num=data.peek();
return num;
}
public int min(){
int num=min.peek();
return num;
}
}