如何用O(1)的时间复杂度求栈中最小元素

11 篇文章 0 订阅

由于栈具有后进先出(Last In First Out, LIFO)的特点,因此 push 和 pop 只需要对栈顶元素进行操作,只能访问到栈顶元素,而无法得到栈中最小的元素。那么,如何求栈中最小元素呢?

1. 利用一个变量记录栈底的位置,通过遍历栈中的所有元素找出最小值

但是这种方法的时间复杂度为 O(n)。

那么如何才能用时间复杂度为O(1)求出栈中最小的元素呢?在算法设计中,经常会采用空间来换取时间的方式,也就是说,采用额外的存储空间来降低操作的时间复杂度。下面我们将介绍一种用时间复杂度为O(1)求栈中最小元素的方法。

2. 使用两个栈结构,一个栈用来存储数据,另一个栈用来存储栈的最小元素

实现思路如下:

(1)使用 elem 和 min 两个栈结构,elem 用来存储数据,min 用来存储 elem 栈的最小元素。

(2)在入栈时,如果当前入栈的元素比原来栈中的最小值还小,则把这个值压入 min 栈中;

(3)在出栈时,如果当前出栈的元素巧好为当前栈中的最小值,则 min 栈的栈顶元素也出栈,是的当前最小值变为其入栈之前的那个最小值。

为了简单起见,可以在栈中保存 Integer 类型,采用链表的方式实现栈,实现代码如下:

 public class GetStackMinEle {
    MyStack<Integer> elem;
    MyStack<Integer> min;

    public GetStackMinEle() {
        elem = new MyStack<Integer>();
        min = new MyStack<Integer>();
    }
    // 入栈时需要考虑当前元素是否是当前栈最小元素,是的话当前元素也要压入 min 栈中
    public void push(int data) { 
        elem.push(data);
        if(min.isEmpty())
            min.push(data);
        else {
            if(data < min.peek())
                min.push(data);
        }
    }
    //出栈时,需要判断 elem 栈顶元素是否为当前栈的最小值,是的话,min 栈顶元素也要出栈
    public int pop() {
        int topData = elem.peek();
        elem.pop();
        if(topData == this.min())
            min.pop();
        return topData;
    }

    public int min() {
        if(min.isEmpty())
            return Integer.MAX_VALUE;
        else
            return min.peek();
    }
}

假设 elem 栈中元素为 6, 1, 5, 3, 2,那么 min 栈中对应的元素为 1, 2, 3。对应的图示如下(红色划线代表出栈):

这里写图片描述

如上图所示, elem 栈顶元素 6 出栈时,由于当前栈的最小值是 1,所以 min 栈元素不动。elem 栈顶元素变为 1。
当 1 出栈时,由于 1 是当前栈的最小元素,所以 min 栈的栈顶元素 1 也要出栈。
以此类推。

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值