码迷,mamicode.com
首页 > 其他好文 > 详细

刷题记录 leetcode155:最小值栈

时间:2020-07-22 20:49:40      阅读:77      评论:0      收藏:0      [点我收藏+]

标签: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(); } }

 

刷题记录 leetcode155:最小值栈

标签:stack   bsp   class   值栈   记录   时间   leetcode   nbsp   style   

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

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!