双栈实现最小栈

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

实现 MinStack 类:

MinStack() 初始化堆栈对象。
void push(int val) 将元素val推入堆栈。
void pop() 删除堆栈顶部的元素。
int top() 获取堆栈顶部的元素。
int getMin() 获取堆栈中的最小元素。
 

示例 1:

输入:
["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.
链接:https://leetcode-cn.com/problems/min-stack

解题思路如下:

        设立两个栈S1,S2,前者为存储元素的栈,后者为辅助栈。将-2,  0,-3 依次进栈找出最小元素。当进入第二个元素0时,在S2中比较将要入栈的数和栈内数的大小,若大该数不入栈,将S2中最小的数再入一次S2栈。若小则入栈。但S1正常顺序入栈。这样是为了保证S1和S2中的数字个数相同且S2栈顶元素为栈内最小值。

 

class MinStack {
      //具体存储元素
      private Stack<Integer> element = new Stack<>();
      //辅助栈
      private Stack<Integer> support = new Stack<>();

    public MinStack() {
      
    }
    
    public void push(int val) {
        element.push(val);
        if(support.isEmpty()){
            support.push(val);
        }else{
            int tmpMin = support.peek();
            //比较当前元素和tmpMin谁更小
            //将最小的元素push进辅助栈中
            int min = Math.min(val,tmpMin);
            support.push(min);
        }

    }
    
    public void pop() {
        element.pop();
        support.pop();

    }
    
    public int top() {
       return element.peek();
    }
    
    public int getMin() {
       return support.peek();
    }
}

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值