LeetCode42. 接雨水

论坛 期权论坛 编程之家     
选择匿名的用户   2021-5-16 23:34   43   0

题目

给定 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

分析

1.自己一开始的想法:把上图分为一层一层,每一层的高度为1,用栈分别计算每层的凹进去格子个数,加在一起,哈哈哈哈哈,数据量小还可以用,数据量一大,就超时了,时间复杂度O(n^2)。

2.还是看了解题报告,只要遍历三次,就可以得出结果,哇塞好神奇。

其实这个题只要能知道第x个柱子的 左边最高柱子 和 右边最高柱子 中比较矮的一个,然后用较矮柱子的高度减去当前第x个柱子的高度,就是当前x柱子可以接住的雨水量。

说的有点儿绕,写成表达式就是: x柱子接住雨水量 = min( left_max, right_max) - height[x]

代码

class Solution {
    public int trap(int[] height) {
        int len = height.length;
        if (len == 0) return 0;
        int[] left_max = new int[len];
        int[] right_max = new int[len];

        int max = height[0];
        for (int i = 0; i < len; i++) {
            if (max > height[i]){
                left_max[i] = max;
            }else{
                left_max[i] = height[i];
                max = height[i];
            }
        }

        max = height[len-1];
        for (int i = len-2; i >= 0; i--) {
            if (max > height[i]){
                right_max[i] = max;
            }else{
                right_max[i] = height[i];
                max = height[i];
            }
        }

        int sum = 0;
        for (int i = 1; i < len-1; i++) {
            sum += Math.min(left_max[i],right_max[i]) - height[i];
        }

        return sum;
    }
}

分享到 :
0 人收藏
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

积分:3875789
帖子:775174
精华:0
期权论坛 期权论坛
发布
内容

下载期权论坛手机APP