最小栈
实现一个最小栈,一步一步优化,从额外空间O(N) 到O(1) 。面试官看重代码逻辑。push
,pop
,top
,getMin
都是O(1)时间。
1 用一个最小栈来存储最小值
1.1要点:
2个栈,data用来存储数据,minValue用来存储最小值。
push
时,data直接push
数据;minValue直接放入当前最小的值。(对于minValue有一个优化,当push
的数据比当前最小值大的时候,我们可以不对minValue进行最小值的插入;如果小于或者等于最小值,就需要把最新的最小值push
入栈minValue。pop
时,data直接pop
出数据;同时,更新minValue,更新的策略是与push
中的优化对应的策略——pop
出的数,如果==当前的最小值,就需要把minValue进行pop
一次。getMin
:直接返回栈minValue 的 top元素即可。top
: 直接返回栈data的top元素即可。
1.2 复杂度和代码
额外空间消耗O(N)
,如何优化到O(1)
.
public class MinStack1 { private Stack<Integer> data = new Stack<Integer>(); private Stack<Integer> minValue = new Stack<Integer>(); public void push(int x) { data.push(x); if (minValue.isEmpty() || x <= minValue.peek()) minValue.push(x); } public void pop() { int value = data.pop(); if (value == minValue.peek()) minValue.pop(); } public int top() { return data.peek(); } public int getMin() { return minValue.peek(); } }
2 优化空间复杂度到O(1)
如何只用一个栈实现最小栈的实现?
栈不能够只存储原始数据,应该存储差值。
用一个变量来计算栈的最小值
用简单的示例来探索思路。
2.1 图
入栈顺序:2,1,3,4,-2,0,-2
diff栈的计算 = data - min
出栈的data | 最小值 | diff栈 | 最小值min | |
---|---|---|---|---|
2 | 2 | 0 | 2 | |
1 | 1 | -1 | 1 | |
3 | 1 | 2 | 1 | |
4 | 1 | 3 | 1 | |
-2 | -2 | -3 | -2 | |
0 | -2 | 2 | -2 | |
-2 | -2 | 0 | -2 |
top
: 如何根据diff栈来恢复栈顶top的元素?push
: 如何更新min最小值?pop
: 如何维护min的最小值?
2.2 代码
注意:第一次入栈diff的特殊处理。
public class MinStack3 { private Stack<Integer> diff = new Stack<Integer>(); private int minValue; public void push(int x) { if (diff.isEmpty()) { minValue = x; diff.push(0); } else { int compare = x - minValue; diff.push(compare); minValue = compare < 0 ? x : minValue; } } public void pop() { int top = diff.peek(); minValue = top < 0 ? (minValue - top) : minValue; diff.pop(); } public int top() { int top = diff.peek(); return top > 0 ? top + minValue : minValue; } public int getMin() { return minValue; } }
转载于:https://blog.51cto.com/zhangtaoze/1912834