题目:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
- MinStack() 初始化堆栈对象。
- void push(int val) 将元素val推入堆栈。
- void pop()删除堆栈顶部的元素。
- int top() 获取堆栈顶部的元素。
- int getMin() 获取堆栈中的最小元素。
示例 1:
输入: [“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]输出: [null,null,null,null,-3,null,0,-2]
解释: MinStack minStack = new MinStack(); minStack.push(-2);
minStack.push(0); minStack.push(-3); minStack.getMin(); --> 返回 -3.
minStack.pop(); minStack.top(); --> 返回 0. minStack.getMin();
–> 返回 -2.
思路
题目要求时间复杂度为O(1),所以需要用空间换时间 构造两个栈A、B,B中的内容,就是由一系列最小元素构成的
每次给A中插入元素,都需要拿A的元素和B的栈顶进行对比,把较小的值插入到B中 后续获取最小值,直接从B的栈顶来取就行了
如果对A出栈,也同步对B出栈,出栈元素后,B的栈顶仍然是A栈的最小值
代码
class MinStack {
Stack<Integer> A = new Stack<>();
Stack<Integer> B = new Stack<>();
public MinStack() {
}
public void push(int val) {
//先把x插入A中
A.push(val);
//比较val和B的栈顶,谁小就插入谁
if(B.isEmpty()){
B.push(val);
return;
}
int min = B.peek();
if(min > val){
min = val;
}
B.push(min);
}
public void pop() {
if(A.isEmpty()){
return;
}
A.pop();
B.pop();
}
public int top() {
if(A.isEmpty()){
return 0;
}
return A.peek();
}
public int getMin() {
if(B.isEmpty()){
return 0;
}
return B.peek();
}
}
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/min-stack
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。