Design a stack that supports push, pop, top, and retrieving the minimum element in constant time. push(x) -- Push element x onto stack. pop() -- Removes the element on top of the stack. top() -- Get the top element. getMin() -- Retrieve the minimum element in the stack.
这是一道关于栈的题目,整体来说比较简单。我最开始想的时候想的太简单,以为使用一个栈st, 再维护一个最小值变量就好了。突然发现pop()操作之后需要更新这个最小值,那就需要知道第二小的值,这个第二小的值怎么找呢?于是乎我想到了使用另外一个栈minst专门来存储最小值。push()操作的时候每当x小于或等于(注意不是小于,之所以小于等于是为了元素重复的情况)minst的栈顶元素,minst也push x入栈。pop()操作时,如果pop出来的元素等于minst栈顶元素,那么minst栈也出栈。
这道题思路很简单,到时语法卡了我一会儿,老是过不了,一段时间不练手生啊。我的语法问题又出在==跟equals()上面。我栈的定义是Stack<Integer>。Integer是一个object。那么判断两个栈栈顶元素相等就不能写成 st.peek() == minst.peek(),这是地址相等,equals才是值相等。要改的话要么就改用equals,要么定义一个int elem = st.pop(); 再判断if (elem == minst.peek()), 这里是一个int 变量跟Integer对象相比,而不是两个Integer对象相比,==在这里就是表示值相等。
1 class MinStack { 2 Stack<Integer> st; 3 Stack<Integer> minst; 4 5 public MinStack() { 6 this.st = new Stack<Integer>(); 7 this.minst = new Stack<Integer>(); 8 } 9 10 public void push(int x) { 11 st.push(x); 12 if (minst.empty() || minst.peek()>=x) { 13 minst.push(x); 14 } 15 } 16 17 public void pop() { 18 if (st.isEmpty()) { 19 return; 20 } 21 if (st.peek().equals(minst.peek())) { 22 minst.pop(); 23 } 24 st.pop(); 25 } 26 27 public int top() { 28 if (!st.isEmpty()) { 29 return st.peek(); 30 } 31 return Integer.MAX_VALUE; 32 } 33 34 public int getMin() { 35 if (!minst.isEmpty()) { 36 return minst.peek(); 37 } 38 return Integer.MAX_VALUE; 39 } 40 }
(Adopted approach) use just one stack:
Push twice when min value changes.
Pop twice as well
1 class MinStack { 2 int min = Integer.MAX_VALUE; 3 Stack<Integer> stack = new Stack<Integer>(); 4 public void push(int x) { 5 // only push the old minimum value when the current 6 // minimum value changes after pushing the new value x 7 if(x <= min){ 8 stack.push(min); 9 min=x; 10 } 11 stack.push(x); 12 } 13 14 public void pop() { 15 // if pop operation could result in the changing of the current minimum value, 16 // pop twice and change the current minimum value to the last minimum value. 17 if(stack.pop() == min) min=stack.pop(); 18 } 19 20 public int top() { 21 return stack.peek(); 22 } 23 24 public int getMin() { 25 return min; 26 } 27 }