标签:stack bsp class 值栈 记录 时间 leetcode nbsp style
设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。
push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() —— 检索栈中的最小元素。
思路:用一个辅助栈的栈顶记录原始栈的最小值,并且两个栈的高度保持相同,入栈时,当前值与辅助栈的栈顶比较,将两者的最小值入栈,这样能保证,在原始栈的最小值弹出前,辅助栈一直保存着最小值。
class MinStack { private Stack<Integer> dataStack; private Stack<Integer> minStack; private int min; public MinStack() { dataStack = new Stack<>(); minStack = new Stack<>(); min = Integer.MAX_VALUE; } public void push(int x) { dataStack.add(x); min = Math.min(min, x); minStack.add(min); } public void pop() { dataStack.pop(); minStack.pop(); min = minStack.isEmpty() ? Integer.MAX_VALUE : minStack.peek(); // 这句话是为了防止两种情况:1. minStack弹出后栈为空,
min必须更新为最大整数,否则min会保持原来栈的最小值,产生错误。
2. 如果弹出后栈不为空,但是弹出的恰好是当前的最小值,
如果min不更新为栈顶元素(新栈的最小值),也会产生错误 } public int top() { return dataStack.peek(); } public int getMin() { return minStack.peek(); } }
标签:stack bsp class 值栈 记录 时间 leetcode nbsp style
原文地址:https://www.cnblogs.com/tendermelon/p/13362481.html