包含min函数的栈(剑指offer)
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
解法:通过两个栈来实现这个栈,一个是正常的数据栈data,另一个是存储最小数据的栈min,当push()的时候,如果min为空,min也直接push();接着每有一个数据入栈data,判断该数与min的栈顶,如果小于等于min.peek(),min也push()。当要从栈中pop()一个值的时候,判断该值是否与min.peek()相等,如果相等,min也pop()。这样就保证了min中的栈顶永远都是栈data中最小的元素,因为每有一个更小的进来min才push(),当该值出去时,min也pop(),但min中仍然保存着之前最小的值。
import java.util.Stack;
public class Solution {
Stack<Integer> data = new Stack<Integer>();
Stack<Integer> min = new Stack<Integer>();
public void push(int number){
data.push(number);
if(min.empty() || number<=min.peek()){
min.push(number);
}
}
public int pop(){
if(min.peek()==data.peek()){
min.pop();
}
return data.pop();
}
public int peek(){
return data.peek();
}
public int min(){
return min.peek();
}
}