题目来源:https://leetcode-cn.com/problems/min-stack/
大致题意:
实现一个最小栈 MinStack 类:
- MinStack() 初始化堆栈对象
- void push(int val) 将元素val推入堆栈
- void pop() 删除堆栈顶部的元素
- int top() 获取堆栈顶部的元素
- int getMin() 获取堆栈中的最小元素
思路
题目要求 getMin() 方法复杂度为常数级,所以需要在出入栈的时候维护最小值,可以使用一个辅助栈,栈顶元素为原数栈每个位置对应的栈中最小值
入栈:
- 每次 val 入栈时,取辅助栈当前栈顶元素与 val 的最小值入辅助栈
- 初始时,先给辅助栈放入一个整形的最大值做哨兵
出栈: - 原始栈和辅助栈弹出栈顶元素
获取最小值 - 返回辅助栈栈顶元素
public class MinStack {
Stack<Integer> stack;
Stack<Integer> minStack; // 辅助栈
public MinStack() {
stack = new Stack<>();
minStack = new Stack<>();
// 先放入一个整型最大值
minStack.push(Integer.MAX_VALUE);
}
public void push(int val) {
stack.push(val);
// 将当前最小值放入辅助栈
minStack.push(Math.min(minStack.peek(), val));
}
public void pop() {
stack.pop();
minStack.pop();
}
public int top() {
return stack.peek();
}
public int getMin() {
return minStack.peek();
}
}