LeetCode 42. 接雨水

作者:Choi Yang

更新于:6 个月前

字数统计:353 字

阅读时长:1 分钟

阅读量:

题目描述

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

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

解题思路

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

以左边为例：当前柱子存水量 = 最近最高柱子高度（只看左边到当前柱子） - 当前柱子高度

右边同理。

/**
 * @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;
};

学如逆水行舟，不进则退

Contributors

Choi Yang