剑指 Offer 30. 包含min函数的栈
题目
定义栈的数据结构,请在该类型中实现一个能够得到栈的最小元素的 min 函数在该栈中,调用 min、push 及 pop 的时间复杂度都是 O(1)。
示例:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.min(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.min(); --> 返回 -2.
提示:
各函数的调用总次数不超过 20000 次
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/bao-han-minhan-shu-de-zhan-lcof
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
前置知识:
标签:
思路
1.自己的思路(栈+ 优先队列,未通过全部案例)
使用优先队列维持最小元素,但是入栈的时候需要额外考虑优先队列的复杂度,无法做到O(1)复杂度。
代码
class MinStack {
private PriorityQueue<Integer> priorityQueue;
private Deque<Integer> stack;
/**
* initialize your data structure here.
*/
public MinStack() {
this.priorityQueue = new PriorityQueue<>((o1, o2) -> (o1 - o2));
this.stack = new LinkedList<>();
}
public void push(int x) {
priorityQueue.add(x);
stack.push(x);
}
public void pop() {
if (!stack.isEmpty()) {
int peek = stack.pop();
priorityQueue.remove(peek);
}
}
public int top() {
return stack.isEmpty() ? -1 : stack.peek();
}
public int min() {
return priorityQueue.isEmpty() ? -1 : priorityQueue.peek();
}
}
2.题解思路
普通栈的 push()
和 pop()
函数的复杂度为 O(1) ;而获取栈最小值 min()
函数需要遍历整个栈,复杂度为 O(N)。
将min()
函数复杂度降为O(1),可通过建立辅助栈实现;
- 数据栈 A : 栈 A 用于存储所有元素,保证入栈
push()
函数、出栈pop()
函数、获取栈顶top()
函数的正常逻辑。 - 辅助栈 B: 栈 B 中存储栈 A 中所有 非严格降序 的元素,则栈 A 中的最小元素始终对应栈 B 的栈顶元素,即
min()
函数只需返回栈 B 的栈顶元素即可。
函数设计:
push(x)
函数: 重点为保持栈 B 的元素是 非严格降序 的。- 将 x 压入栈 A (即
A.add(x)
); - 若 ① 栈 B 为空 或 ② x 小于等于 栈 B 的栈顶元素,则将 x 压入栈 B(即
B.add(x)
)。
- 将 x 压入栈 A (即
pop()
函数: 重点为保持栈 A, B 的 元素一致性 。- 执行栈 A 出栈(即
A.pop()
),将出栈元素记为 y ; - 若 y 等于栈 B 的栈顶元素,则执行栈
B
出栈(即B.pop()
)。
- 执行栈 A 出栈(即
top()
函数: 直接返回栈 A 的栈顶元素即可,即返回A.peek()
。min()
函数: 直接返回栈 B 的栈顶元素即可,即返回B.peek()
。
代码
class MinStack {
private Deque<Integer> stack;
private Deque<Integer> minValue;
/**
* initialize your data structure here.
*/
public MinStack() {
this.stack = new LinkedList<>();
this.minValue = new LinkedList<>();
}
public void push(int x) {
stack.push(x);
if (minValue.isEmpty() || minValue.peek() >= x){
minValue.push(x);
}
}
public void pop() {
if (!stack.isEmpty()) {
if (!minValue.isEmpty() && minValue.peek().equals(stack.pop())){
minValue.pop();
}
}
}
public int top() {
return stack.isEmpty() ? -1 : stack.peek();
}
public int min() {
if (minValue.isEmpty()){
if (stack.isEmpty()){
return -1;
}else {
return stack.peek();
}
}
return minValue.peek();
}
}
复杂度分析
- 时间复杂度:
push()
,pop()
,top()
,min()
四个函数的时间复杂度均为常数级别。 - 空间复杂度:当共有 n 个待入栈元素时,辅助栈 minValue 最差情况下存储 n个元素,使用 O(n) 额外空间。
总结
对于实现栈的题目,优先思考使用栈本身是否可以解决问题,不行再考虑其他解法