包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
思路
/*
*
* 借助一个栈 minstack 来储存最小元素
* 每次入栈 判断最小栈顶元素的值是否 小于等于入栈的值如果满足这个条件这吧这个元素压入到最小栈中。
* 为什么是等于最小栈顶元素的也要压入 :因为在pop的时候会判断当前pop的元素是不是最小值,如果是这最小栈也要pop
* 如果存在重复的最小值而没有被重复压入。则在pop的时候会把最小值出栈,则最小栈顶元素就不是当前最小的值。所以需要重复压入。
*
* 时间复杂度o(1)
* 空间复杂复杂度o(n) 最坏情况所有元素都需要压入到最小栈中
* */
代码
public class Offer30 {
private Stack<Integer> stack;
private Stack <Integer> minStack;
/** initialize your data structure here. */
public Offer30() {
stack = new Stack<>();
minStack = new Stack<>();
}
public void push(int x) {
stack.push(x);
if(minStack.isEmpty()){ //如果最小栈为空 则插入x 即可
minStack.push(x);
} else if(x <= minStack.peek()) { //如果x 比minstack栈顶元素还小则压入到minstack中
minstack.push(x);
}
}
public void pop() {
if (stack.peek().equals(min())) {
stack.pop();
minStack.pop();
} else {
stack.pop();
}
}
public int top() {
return stack.peek();
}
public int min() {
return minStack.peek();
}
}