接雨水

80 阅读1分钟

题目

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

示例

309f802cf9e64e06a4e97c7a3abb3b99.png

输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]

输出:6

解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单

位的雨水(蓝色部分表示雨水)。

代码实现

/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function (height) {
  var rain = 0;
  for (let i = 1; i < height.length - 1; i++) {
    let max_left = 0;
    for (let j = i - 1; j >= 0; j--) {
      if (height[j] > max_left) {
        max_left = height[j];
      }
    }
    let max_right = 0;
    for (let j = i + 1; j < height.length; j++) {
      if (height[j] > max_right) {
        max_right = height[j];
      }
    }
    let min = Math.min(max_left, max_right);
    if (min > height[i]) {
      rain = rain + (min - height[i]);
    }
  }
  return rain;
};