剑指offer-test20

20.包含min函数的栈
定义栈的数据结构,请在该类型中实现一个能够得到栈中所含最小元素的min函数(时间复杂度应为O(1))。
算法描述:
1.定义两个栈,一个是存传入的数据栈dataStack,一个是存储最小元素的栈minStack。
2.push操作:每次传入的数据node都push进dataStack栈中。然后判断minStack栈是否为空,如果为空则把node数据push进minStack栈中;如果不为空,就将传入的数据node和minStack中栈顶的元素比较,如果node较小或等于,则push进minStack,否则就不压进栈minStack。
3.pop操作:当数据栈要弹出元素时!!!为了保证辅助栈的栈顶元素始终是最小值,需要对辅助栈同时进行一个弹栈操作。但是首先判断两个栈的栈顶元素是否相等,相等话则辅助栈才执行弹栈操作,否则最小栈不用执行弹栈操作。
下图展示了栈内压入3、4、2、1之后接连两次弹出栈顶数字再压入0时,数据栈、辅助栈和最小值的状态。
在这里插入图片描述
例如:data中依次入栈,5, 4, 3, 8, 10, 11, 12, 1
则min依次入栈,5, 4, 3,no,no, no, no, 1,依次找次小值,只能找到有限个,并且与入栈元素顺序有关。

import java.util.Stack;
public class Solution {
    Stack<Integer> dataStack =new Stack<Integer>();
    Stack<Integer> minStack =new Stack<Integer>();
    public void push(int node) {
        dataStack.push(node);
        if(minStack.isEmpty()){
            minStack.push(node);
        }
        else if(node<=minStack.peek()){
          //peek函数是取minstack的栈顶元素。
            minStack.push(node);
        }
    }
    public void pop(){
            if(dataStack.peek() == minStack.peek())
                minStack.pop();   
             dataStack.pop(); 
    }
    public  int min(){
         return minStack.peek();
    }
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值