题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的min函数。在该栈中,调用min,push及pop的时间复杂度都是O(1)。
解
import java.util.Stack;
public class StackWithMin {
Stack<Integer> m_data=new Stack<Integer>();//数据栈
Stack<Integer> m_min=new Stack<Integer>();//辅助栈,用来存储当前数据集对应的最小元素
void push(Integer value){
m_data.push(value);
if(m_min.size()==0||value<m_min.peek()){
m_min.push(value);
}else{
m_min.push(m_min.peek());
}
}
void pop(){
if(m_min.size()>0 && m_data.size()>0){
m_data.pop();
m_min.pop();
}
}
Integer min(){
if(m_min.size()>0 && m_data.size()>0){
return m_min.peek();
}else{
return null;
}
}
public static void main(String[] args) {
// TODO Auto-generated method stub
}
}