Leetcode: Min Stack

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 }

 

你可能感兴趣的:(LeetCode)