最小栈的实现

题目: 实现一个栈,该栈带有出栈(pop)、入栈(push)、取最小元素(getMin)3个方法,要保证这3个方法的时间复杂度都是O(1)

解题思路:

  1. 设原有的栈叫做栈A,此时创建一个额外的“备胎”栈B,用于辅助栈A;
  2. 当第一个元素进入栈A,让新元素也进入栈B。这个唯一的元素就是栈A的当前最小值。
  3. 之后每当新元素进入栈A时,比较新元素和栈A当前最小值的大小,如果小于栈A当前最小值,则让新元素进入栈B,此时栈B的栈顶元素就是栈A当前最小值;
  4. 每当栈A有元素出栈时,如果出栈元素是栈A当前最小值,则让栈B的栈顶元素也出栈。此时栈B余下的栈顶元素所指向的是栈A当中原本第2小的元素,代替刚才出栈的元素成为最小值。

代码实现:

	private Stack<Integer> mainStack=new Stack<>();//主栈
    private Stack<Integer> minStack=new Stack<>();//辅助栈

    /**
     * 入栈操作
     * @param element 入栈的元素
     */
    public void push(int element){
        mainStack.push(element);
        //如果辅助栈为空,或者新元素小于或等于辅助栈栈顶,则将新元素压入辅助栈
        if (minStack.empty()||element<=minStack.peek()){
            minStack.push(element);
        }
    }

    //出栈操作
    public Integer pop(){
        //如果出栈元素和辅助栈顶元素值相等,辅助栈出栈
        if (mainStack.peek().equals(minStack.peek())){
            minStack.pop();
        }
        return mainStack.pop();
    }

    //获取栈中的最小元素
    public int getMin() throws Exception {
        if (mainStack.empty()){
            throw new Exception("stack is empty");
        }

        return minStack.peek();
    }
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值