【LeetCode刷题】最小栈(Java)

设计一个支持 push ,pop ,top 操作,并能在常数时间内检索到最小元素的栈。

push(x) —— 将元素 x 推入栈中。
pop() —— 删除栈顶的元素。
top() —— 获取栈顶元素。
getMin() ——检索栈中的最小元素。

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/min-stack

示例:

输入:
[“MinStack”,“push”,“push”,“push”,“getMin”,“pop”,“top”,“getMin”]
[[],[-2],[0],[-3],[],[],[],[]]

输出:
[null,null,null,null,-3,null,0,-2]

解释:
MinStack minStack = new MinStack();
minStack.push(-2);
minStack.push(0);
minStack.push(-3);
minStack.getMin(); --> 返回 -3.
minStack.pop();
minStack.top(); --> 返回 0.
minStack.getMin(); --> 返回 -2.

提示:
pop、top 和 getMin 操作总是在 非空栈 上调用。

题目分析:
本题要求常数时间内,所以不能让所有元素进栈后再遍历,可以用一个辅助栈来解决。创建两个栈st和minst,st用来使所有元素入栈,minst中使当前最小值元素入栈。每个元素压入st时,与minst中栈顶元素作比较,得到当前最小元素,将最小元素压入minst,如果最小元素值不变就不压入minst。当要弹出栈中元素时,首先弹出st栈中的栈顶元素,如果栈顶元素是当前最小值元素,也将minst栈顶元素弹出。如果要获取当前栈栈顶元素,根据题意获取的是st的栈顶元素,返回st的栈顶元素即可。要获取当前栈中最小值,返回minst中的栈顶元素即可。
因为pop、top 和 getMin 操作总是在 非空栈 上调用,所以不用判断栈是否为空。

要注意的是,将元素压入st中时,与minst中栈顶元素作比较条件是小于等于,不是小于,参考全部压入相同元素的例子,如果相同时不压入,弹出栈顶元素时就会出错。

class MinStack {
    Stack<Integer> st;
    Stack<Integer> minst;
  
    public MinStack() {
        st=new Stack<>();
        minst=new Stack<>();
    }
    
    public void push(int x) {
        st.push(x);
        if(minst.empty() || x<=minst.peek()) minst.push(x);//如果minst为空,说明x为第一个元素,即为当前最小值。
        //注意判断条件是小于等于
    }
    
    public void pop() {
        int t=st.peek();
        st.pop();
        if(t==minst.peek()) minst.pop();
    }
    
    public int top() {
        return st.peek();
    }
    
    public int getMin() {
        return minst.peek();
    }
}
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值