码迷,mamicode.com
首页 > 其他好文 > 详细

Leetcode: Min Stack

时间:2014-11-22 07:04:11      阅读:193      评论:0      收藏:0      [点我收藏+]

标签:des   style   blog   color   使用   sp   on   div   问题   

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: Min Stack

标签:des   style   blog   color   使用   sp   on   div   问题   

原文地址:http://www.cnblogs.com/EdwardLiu/p/4114642.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!