LeetCode第155题 最小栈

设计一个支持 push,pop,top 操作,并能在常数时间内检索到最小元素的栈。

push(x) -- 将元素 x 推入栈中。
pop() -- 删除栈顶的元素。
top() -- 获取栈顶元素。
getMin() -- 检索栈中的最小元素。
示例:

MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.


思路: 使用太多封装虽然偷了懒,效率影响确实很大...

使用两个栈,一个保存真实的数据,另一个保存所有最小值

eg:
push : -1 0 -2 3 4
push操作完成后,两个栈的情况如下:
stack: -1 0 -2 3 4
minData: -1 -2

只要当前mindata的top(-2)未被弹出,栈中的最小值就一直是-2.
如果此时stack弹出的值为(-2),那么mindata中的值也要更新,将-2弹出.
那么此时
stack: -1 0
minData: -1
栈中的最小值就为-1了


 1 class MinStack {
 2 
 3   /**
 4    * initialize your data structure here.
 5    */
 6   private Stack stack;     //
 7   private Stack minData;   //保存最小值的栈
 8 
 9   public MinStack() {
10 
11     stack = new Stack<>();
12     minData = new Stack<>();
13   }
14 
15   public void push(int x) {
16     stack.push(x);
17     if (minData.empty() || minData.peek() >= x) {
18       minData.push(x);
19     }
20   }
21 
22   public void pop() {
23     if (stack.peek().equals(minData.peek())) {
24       minData.pop();
25     }
26     stack.pop();
27   }
28 
29   public int top() {
30     return stack.peek();
31   }
32 
33   public int getMin() {
34     return minData.peek();
35   }
36 }

 

转载于:https://www.cnblogs.com/rainbow-/p/10390860.html

你可能感兴趣的:(LeetCode第155题 最小栈)