LeetCode42. 接雨水

最新推荐文章于 2024-01-16 01:42:41 发布

一点一滴g

最新推荐文章于 2024-01-16 01:42:41 发布

阅读量81

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/qq_38311489/article/details/89819875

算法专栏收录该内容

65 篇文章

订阅专栏

博客给出一个计算柱子排列下雨后可接雨水量的题目，如数组 [0,1,0,2,1,0,1,3,2,1,2,1] 可接 6 个单位雨水。题解思路是使用两根指针，一根从右往左记录右边峰值，一根从左往右更新左边峰值并计算积水，水面高度取左右峰值最小值。

题目描述

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
在这里插入图片描述
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

题解

思路
使用两根指针：

一根从右往左，更新记录从位置i往右边的峰值;
一根从左往右遍历。更新从位置i往左边峰值的同时，计算位置i处可以积水多少;
水面高度h应为min(maxL,maxR)。积水应为max(0,h-height[I]);

代码：

class Solution {
    public int trap(int[] height) {
        if(height == null || height.length <= 2)
            return 0;
        int maxL = height[0];
        int[] maxRs = new int[height.length];
        int waterSum = 0;//计算总的水量
        int maxR = 0;
        for(int i = height.length - 1; i >= 0; i--){
            if(height[i] > maxR) {
                maxRs[i] = maxR = height[i];
            }
            else {
                maxRs[i] = maxR;
            }
        }
        for(int i = 1; i < height.length - 1; i++){
            if(height[i] > maxL) {
                maxL = height[i];//更新左边最大值
            }
                waterSum += Math.max(Math.min(maxL, maxRs[i]) - height[i], 0);
        }
        return waterSum;
        
      }

}