最小栈

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

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

思路一:
本题的关键在于在常数时间内检索到最小元素。所以要用空间换时间,思路一采用辅助栈的方式来做。
主栈存储数据,辅助栈的栈顶存储当前栈内的最小元素。

辅助栈的具体操作流程如下:

将第一个元素入栈。

新加入的元素如果大于栈顶元素,那么新加入的元素就不处理。

新加入的元素如果小于等于栈顶元素,那么就将新元素入栈。

出栈元素不等于栈顶元素,不操作。

出栈元素等于栈顶元素,那么就将栈顶元素出栈。

class MinStack {
    private Stack<Integer> main;
    private Stack<Integer> min;
    /** initialize your data structure here. */
    public MinStack() {
        main = new Stack<Integer>();
        min = new Stack<Integer>();
    }
    
    public void push(int x) {
        main.push(x);
        if(min.empty() || x<=min.peek()){   //特别注意:这里等于的情况下也要进栈,否则会出错
            min.push(x);
        }
    }
    
    public void pop() {
        int n = main.pop();
        if(n==min.peek()){
            min.pop();
        }
    }
    
    public int top() {
        return main.peek();
    }
    
    public int getMin() {
        return min.peek();
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */

思路二:
思路一中,使用一个栈来保存最小值。可不可以使用一个int变量来保存最小值呢?

入栈 3 
|   |   min = 3
|   |     
|_3_|    
stack   

入栈 5 
|   |   min = 3
| 5 |     
|_3_|    
stack  

入栈 2 
| 2 |   min = 2?
| 5 |     
|_3_|    
stack  

但是这样一来,原来的最小值3就丢失了。解决办法:只需要在2入栈的时候先将3入栈即可。

入栈 2 ,同时将之前的 min 值 3 入栈,再把 2 入栈,同时更新 min = 2
| 2 |   min = 2
| 3 |  
| 5 |     
|_3_|    
stack  

入栈 6 
| 6 |  min = 2
| 2 |   
| 3 |  
| 5 |     
|_3_|    
stack  

出栈 6     
| 2 |   min = 2
| 3 |  
| 5 |     
|_3_|    
stack  

出栈 2     
| 2 |   min = 2
| 3 |  
| 5 |     
|_3_|    
stack  

这样,当我们出栈2时,将它下面的3也出栈,然后将min更新为3即可。

class MinStack {
    private Stack<Integer> main;
    private int min;
    /** initialize your data structure here. */
    public MinStack() {
        main = new Stack<Integer>();
        min = Integer.MAX_VALUE;
    }
    
    public void push(int x) {
        if(x<=min){   //特别注意:这里等于的情况下也要进栈,否则会出错
            main.push(min); 
            min = x;      
        }
        main.push(x);
    }
    
    public void pop() {
        int n = main.pop();
        if(n==min){
            min = main.pop();
        }
    }
    
    public int top() {
        return main.peek();
    }
    
    public int getMin() {
        return min;
    }
}

/**
 * Your MinStack object will be instantiated and called as such:
 * MinStack obj = new MinStack();
 * obj.push(x);
 * obj.pop();
 * int param_3 = obj.top();
 * int param_4 = obj.getMin();
 */
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值