其他分享
首页 > 其他分享> > 刷题记录 leetcode155:最小值栈

刷题记录 leetcode155:最小值栈

作者:互联网

设计一个支持 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,minStack,min,pop,最小值,dataStack,public,刷题
来源: https://www.cnblogs.com/tendermelon/p/13362481.html