定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 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.
提示:
各函数的调用总次数不超过 20000 次
注意:本题与主站 155 题相同:https://leetcode-cn.com/problems/min-stack/
class MinStack {
ArrayList<Integer> list1 = null;//用来存放数据
ArrayList<Integer> list2 = null;//最来存放最小值
/** initialize your data structure here. */
public MinStack() {
list1 = new ArrayList<>();
list2 = new ArrayList<>();
}
public void push(int x) {
list1.add(x);
if(list2.isEmpty() || x<= list2.get(list2.size()-1) ){
list2.add(x);
}
}
public void pop() {
if(!list1.isEmpty()){
if(list1.get(list1.size()-1).equals(list2.get(list2.size()-1))){//包装类要使用 equals 判断,不能用 ==
list1.remove(list1.size()-1);
list2.remove(list2.size()-1);
} else{
list1.remove(list1.size()-1);
}
}
}
public int top() {
return list1.get(list1.size()-1);
}
public int min() {
return list2.get(list2.size()-1);
}
}