首页 > 其他 > 详细

刷题记录 leetcode155:最小值栈

时间:2020-07-22 21:11:31      阅读:63      评论:0      收藏:0      [点我收藏+]

设计一个支持 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(); } }

 

刷题记录 leetcode155:最小值栈

原文:https://www.cnblogs.com/tendermelon/p/13362481.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!