包含min函数的栈(剑指offer)

包含min函数的栈(剑指offer)


定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。

解法:通过两个栈来实现这个栈,一个是正常的数据栈data,另一个是存储最小数据的栈min,当push()的时候,如果min为空,min也直接push();接着每有一个数据入栈data,判断该数与min的栈顶,如果小于等于min.peek(),min也push()。当要从栈中pop()一个值的时候,判断该值是否与min.peek()相等,如果相等,min也pop()。这样就保证了min中的栈顶永远都是栈data中最小的元素,因为每有一个更小的进来min才push(),当该值出去时,min也pop(),但min中仍然保存着之前最小的值。

import java.util.Stack;

public class Solution {
Stack<Integer> data = new Stack<Integer>();
Stack<Integer> min = new Stack<Integer>();

public void push(int number){
    data.push(number);
    if(min.empty() || number<=min.peek()){
        min.push(number);
    }
}
public int pop(){
    if(min.peek()==data.peek()){
        min.pop();
    }
    return data.pop();
}
public int peek(){
    return data.peek();
}
public int min(){
    return min.peek();
}
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值