题目:
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
实现 MinStack 类:
MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/min-stack
思路:
核心思路就是创建一个栈S2,让S2只存储扫描的最小元素值;
当S2为空的时候,直接入栈;
当S2栈顶元素 > 当前扫描元素时,也直接入栈;
当S2栈顶元素 < 当前扫描元素时,让当前栈顶元素再一次入栈。(原因:此时S2中的元素个数一定要与S1中个数保持一致。因为我么们要保持S1和S2的压栈和入栈操作同步)
动态图解:
代码:
public class Num155_MinStack {
//保存栈实际存储的元素
private Stack<Integer> s1 = new Stack<>();
//保存栈顶中最小值元素
private Stack<Integer> s2 = new Stack<>();
public void push(int val) {
s1.push(val);
if (s2.isEmpty()) {
s2.push(val);
} else {
int peek = s2.peek();
s2.push(Math.min(val, peek));
}
}
public void pop() {
s1.pop();
s2.pop();
}
public int top() {
return s1.peek();
}
public int getMin() {
return s2.peek();
}
}