Leecode 42.接雨水详细解题思路及算法思路

Leecode 42.接雨水详细解题思路及算法思路

题目

• 给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。
• 示例:
  • [外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-Bj8Dt0fb-1665834959093)(D:\load\网页捕获_14-10-2022_211641_leetcode.cn.jpg)]

思路：

• 求每一列的水，我们只需要关注当前列，以及左边最高的墙，右边最高的墙就够了。

  装水的多少，当然根据木桶效应，我们只需要看左边最高的墙和右边最高的墙中较矮的一个就够了。所以，根据较矮的那个墙和当前列的墙的高度可以分为三种情况

  • 首先用两个数组，max_left [i] 代表第 i 列左边最高的墙的高度，max_right[i] 代表第 i 列右边最高的墙的高度。（一定要注意下，第 i 列左（右）边最高的墙，是不包括自身的）

    • 对于 max_left我们其实可以这样求：

      max_left [i] = Max(max_left [i-1],height[i-1])。它前边的墙的左边的最高高度和它前边的墙的高度选一个较大的，就是当前列左边最高的墙了。

    • 对于 max_right我们可以这样求。

      max_right[i] = Max(max_right[i+1],height[i+1]) 。它后边的墙的右边的最高高度和它后边的墙的高度选一个较大的，就是当前列右边最高的墙了。

  • 这样，我们再利用解法二的算法，就不用在 for 循环里每次重新遍历一次求 max_left 和 max_right 了

代码实现：

• 报超时了

• class Solution {
      public int trap(int[] height) {
          int sum = 0;
          int[] max_left=new int[height.length];
          int[] max_right=new int[height.length];
          for (int i = 1; i < height.length - 1; i++) {
              int max_left = 0;
              for (int j = i - 1; j >= 0; j--) {
                  if (height[j] > max_left) {
                      max_left = height[j];
                  }
              }
              int max_right = 0;
              for (int j = i + 1; j < height.length; j++) {
                  if (height[j] > max_right) {
                      max_right = height[j];
                  }
              }
              int min = Math.min(max_left, max_right);
              if (min > height[i]) {
                  sum = sum + (min - height[i]);
              }
          }
          return sum;
      }
  }
  
  

• 优化版

class Solution {
    public int trap(int[] height) {
        int sum = 0;
        int[] max_left=new int[height.length];
        int[] max_right=new int[height.length];
        for (int i = 1; i < height.length - 1; i++)
            max_left[i]=Math.max(max_left[i-1],height[i-1]);

        for(int i=height.length-2;i>=0;i--)
            max_right[i]=Math.max(max_right[i+1],height[i+1]);

        for(int i=1;i<height.length-1;i++){
            int min=Math.min(max_left[i],max_right[i]);
            if(min>height[i])
                sum=sum+min-height[i];
        }          
        return sum;
    }
}