leetcode42-接雨水-栈实现

题目不说了,看上一篇,这一篇是用栈实现的

public static int trap(int[] height) {
        int sum = 0;
        if (height.length == 0) {
            return 0;
        }
        Stack<Integer> stack = new Stack<>();
        //对数组中每一个元素遍历
        for (int i = 0; i <= height.length - 1; i++) {
            int x = 0;
            //当当前元素大于栈顶元素时取出栈顶元素计算
            //如果栈顶的剩余元素的栈顶元素还是比当前元素height[i]小还要继续取然后计算
            while (!stack.empty() && height[i] > height[stack.peek()]) {
                //取出当前栈的栈顶元素
                int last = height[stack.peek()];
                //弹出元素
                stack.pop();
                if (stack.empty()) { // 栈空就出去
                    break;
                }
                //取出的last和左右两面墙的最低求差
                x = Math.min(height[stack.peek()], height[i]) - last;
                //求last和当前元素的距离用来相乘
                //为什么要求x?比如第一次取栈顶是1,则两个数组距离是1,如果是第二次遍历距离就是2了
                int distinct = i - stack.peek() - 1;
                sum += x * distinct;
            }
            //将当前的下表存入栈中,注意是指针,为什么用指针,因为stack只有取栈顶的方法,而本次需要取栈第二高的数,所以存下标
            stack.push(i);
        }
        return sum;
    }

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值