42. 接雨水


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

示例 1:

  • 输入: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 个单位的雨水(蓝色部分表示雨水)

双指针解法

按照列来计算的话,宽度一定是1了,我们再把每一列的雨水的高度求出来就可以了。

可以看出每一列雨水的高度,取决于,该列 左侧最高的柱子和右侧最高的柱子中最矮的那个柱子的高度

即: min(lHeight, rHeight) - height

class Solution {
    public int trap(int[] height) {
        return process(height);
    }

    private int process(int[] height){
        //按列计算
        //i:left Max. rightMax. h= min(leftMax,rightMax)-heigt[i]>0 sum+=h
        if(height.length<=2){
            return 0;
        }
        int sum=0;
        for(int i=1;i0){
                sum+=h;
            }
            
        }
        return sum;
    }

    private int getRangeMax(int[] height,int L,int R){
        int max=0;
        for(int i=L;i<=R;i++){
            max=Math.max(max,height[i]);
        }
        return max;
    }
}

动态规划解法

为了得到两边的最高高度,使用了双指针来遍历,每到一个柱子都向两边遍历一遍,这其实是有重复计算的。

我们把每一个位置的左边最高高度记录在一个数组上(maxLeft),右边最高高度记录在一个数组上(maxRight)。这样就避免了重复计算,这就用到了动态规划。

当前位置,左边的最高高度是前一个位置的左边最高高度和本高度的最大值。

即从左向右遍历:maxLeft[i] = max(height[i], maxLeft[i - 1]);

从右向左遍历:maxRight[i] = max(height[i], maxRight[i + 1]);

这样就找到递推公式。

class Solution {
    public int trap(int[] height) {
        return process(height);
    }

    private int process(int[] height){
       //leftMax[i] 0....i 求得每个位置左边的最大值 leftMax[i]=Math.max(dp[i-1],height[i])
       //rightMax[i] i....R 求得每个位置右边的最大值 rightMax[i]=Maht.max(dp[i+1],heigth[i])
       //遍历1....i-1. Math.min(leftMax(i),rightMax(i))-heigt[i]
       //h >0 sum+=h
       int n=height.length;
       if(n<=2){
           return 0;
       }
       int[] leftMax=new int[n];
       leftMax[0]=height[0];
       for(int i=1;i=0;i--){
           rightMax[i]=Math.max(rightMax[i+1],height[i]);
       }

       int sum=0;
       for(int i=1;i0) sum+=h;
       }

       return sum;
       
    }


}

单调栈解法

单调栈是按照行方向来计算雨水

从栈头(元素从栈头弹出)到栈底的顺序应该是从小到大的顺序。

因为一旦发现添加的柱子高度大于栈头元素了,此时就出现凹槽了,栈头元素就是凹槽底部的柱子,栈头第二个元素就是凹槽左边的柱子,而添加的元素就是凹槽右边的柱子。

遇到相同高度的柱子怎么办。

遇到相同的元素,更新栈内下标,就是将栈里元素(旧下标)弹出,将新元素(新下标)加入栈中。

例如 5 5 1 3 这种情况。如果添加第二个5的时候就应该将第一个5的下标弹出,把第二个5添加到栈中。

因为我们要求宽度的时候 如果遇到相同高度的柱子,需要使用最右边的柱子来计算宽度。

class Solution {
    public int trap(int[] height) {
        if(height.length<=2){
            return 0;
        }
        return process(height);
    }

    private int process(int[] height){
      //stack bigger.....smaller. when i the bigger then peek(),then pop,
      //i,popi,peeki, Math.min(height[i],hight[stack.peek()])-height[popi]-->h
      //w=i-peeki-1
      //h*w
      int sum=0;
      Stack stack=new Stack<>();
      stack.push(0);
      for(int i=1;iheight[stack.peek()]){
                  int popi=stack.pop();
                  if(!stack.isEmpty()){
                      int h=Math.min(height[i],height[stack.peek()])-height[popi];
                      int w=i-stack.peek()-1;
                      int hold=h*w;
                      if(hold>0) sum+=hold;
                  }
               
              }
              stack.push(i);
          }
      }
      return sum;
    }


}