Skip to content
导航栏

LeetCode 42. 接雨水

题目描述

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。

示例:

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

来源:力扣(LeetCode) 链接:https://leetcode-cn.com/problems/trapping-rain-water 著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

这个存放水,我们就需要看左边两边指针的柱子看谁的高度小了,当前是看高度小的了。

以左边为例:当前柱子存水量 = 最近最高柱子高度(只看左边到当前柱子) - 当前柱子高度

右边同理。

javascript
/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function (height) {
  let len = height.length;
  let L = 0,
    R = len - 1;
  let leftHeight = 0,
    rightHeight = 0;
  let res = 0;
  while (L < R) {
    if (height[L] < height[R]) {
      // 左边高度小,当然看左边
      leftHeight = Math.max(leftHeight, height[L]);
      res += leftHeight - height[L]; // 当前柱子能存放的水
      L++;
    } else {
      // 右边高度小,看右边
      rightHeight = Math.max(rightHeight, height[R]);
      res += rightHeight - height[R]; // 当前柱子能存放的水
      R--;
    }
  }
  return res;
};
javascript
学如逆水行舟,不进则退