定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
难点:min函数要求时间复杂度为O(1)。如果采用简单的遍历,复杂度势必为O(n)
思路:额外创建一个栈。栈A用来进行正常的压入和弹出,栈B用来维护最小值。当栈A压入一个数据时,都会和栈B的栈顶元素进行比较,如果此数据不大于栈B的栈顶元素时,就把此数据压入栈B。当栈A进行弹出操作时,也会进行比较,相同则也将栈B的元素弹出。
class MinStack {
private Stack<Integer> stackA;
private Stack<Integer> stackB;
public MinStack() {
stackA = new Stack<Integer>();
stackB = new Stack<Integer>();
}
public void push(int x) {
stackA.push(x);
if(stackB.empty() || stackB.peek()>=x){
stackB.push(x);
}
}
public void pop() {
if(stackA.pop().equals(stackB.peek())){
stackB.pop();
}
}
public int top() {
return stackA.peek();
}
public int min() {
return stackB.peek();
}
}