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

由于栈具有后进先出的特点,因此push和pop只需要对栈顶元素进行操作。

如果使用上述的实现方式,只能访问到栈顶的元素,而无法得到栈中最小的元素。当然,可以用另外一个变量来记录栈底的位置,通过遍历栈中的所有元素找出最小值,但是这种方法的时间复杂度为O(n)。

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

  在算法设计中,经常会采用空间来换取时间的方式来提高时间复杂度,也就是说,采用额外的存储空间来降低操作的时间复杂度。具体来讲就是在实现时使用两个栈结构,一个栈用来存储数据,另一个栈用来存储栈的最小元素。

其实现思路如下

  如果当前入栈的元素比原来栈中的最小值还小,则把这个值压入保存最小元素的栈中;在出栈时,如果当前出栈的元素恰好为当前栈中的最小值,保存最小的栈顶元素也出栈,使得当前最小值变为其入栈之前的那个最小值。为了简单起见,可以在栈中保存Interger类型,采用前面用链表方式实现的栈,实现代码如下

public class MyStack1{
    MyStack<Integer> elem;
    MyStack<Integer> min;
    public MyStack1(){
        elem = new MyStack <Integer> ();
        min = new MyStack <Integer> ();
    }
    public void push(int data){
      elem.push(data);
      if(min.isEmpty())
          min.push(data);
      else{
           if(data < min.peek())
                min.push(data);
          }
     }
     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();
     }
}

 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值