题目:设计一个栈,该栈支持4种方法
1、push(x)
2、pop()
3、top()
4、getMin() 返回栈中最小元素,要求时间O(1)
题目不难,java中本身就有栈这个数据结构,难点在于O(1)取出栈中最小值。
思路是:
1、设置变量min,每push一个value就更新当前栈中最小值,但是如果该最小值被pop出来后,无法得知下一个最小值。(舍弃)
2、增加一个栈,每次push一个value,就记录当前栈中的最小值,空间为O(n),每当栈pop,最小栈也跟着pop。
3、优化方法2,新增一个栈,但每push一个value,如果该值比最小栈中的最小的元素小得话(或相等)就push到最小栈中,否则直接跳过,pop时判断pop出来的值是否为最小栈中的top值,如果是,最小栈同步pop
注意:方法3中仅存最小值,但有相等时也需存,因为如果仅存一个,如果栈中存在2个最小值的话,pop完一个,最小栈中也pop出,下一次getMin的值就不准确了。
class MinStack {
Stack<Integer> stack;
Stack<Integer> min;
public MinStack(){
stack = new Stack<Integer>();
min = new Stack<Integer>();
}
public void push(int x) {
stack.push(x);
if (min.empty() || min.peek()>x || min.peek() == x) {
min.push(x);
}
}
public void pop() {
if ((int)stack.peek() == (int)min.peek()) {
min.pop();
}
stack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return min.peek();
}
}