力扣刷题(day0024)接雨水

给定 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 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

 输入：height = [4,2,0,3,2,5]
 输出：9

提示：

• n == height.length

• 1 <= n <= 2 * 104

• 0 <= height[i] <= 105

双指针法按列：

class Solution {
 public:
     int trap(vector<int>& height) {
         int sum=0;
         for(int i=0;i<height.size();i++){
             //最前和最后不接雨水
             if(i==0 || i==height.size()-1) continue;
             //定义双指针
             int rheight=height[i];//记录右侧最高
             int lheight=height[i];//记录左侧最高
             //遍历找到i列右侧最高的柱子
             for(int r=i+1;r<height.size();r++){
                 if(height[r]>rheight) rheight=height[r];
             }
             //遍历找到i列左侧最高的柱子
             for(int l=i-1;l>=0;l--){
                 if(height[l]>lheight) lheight=height[l];
             }
             //取左右侧最高柱子中，相对较小的，测量接水最佳高度
             int h=min(lheight,rheight)-height[i];
             if(h>0) sum+=h;
         }
         return sum;
     }
 };

动态规划解法:

 class Solution {
 public:
     int trap(vector<int>& height) {
         if(height.size()<=2) return 0;
         //定义结果变量
         int sum=0;
         //定义两个左右存最大最小值的容器
         vector<int>maxLeft(height.size(),0);
         vector<int>maxRight(height.size(),0);
         //定义一下大小,简洁后面代码
         int size=maxRight.size();
 ​
         //记录每个左边柱子的最大高度
         maxLeft[0]=height[0];
         for(int i=1;i<size;i++){
             maxLeft[i]=max(height[i],maxLeft[i-1]);
         }
         //记录每个右边柱子的最大高度
         maxRight[size-1]=height[size-1];
         for(int i=size-2;i>=0;i--){
             maxRight[i]=max(height[i],maxRight[i+1]);
         }
 ​
         //求和
         for(int i=0;i<size;i++){
             int count=min(maxLeft[i],maxRight[i])-height[i];
             if(count>0)sum+=count;
         }
         return sum;
     }
 };

单调栈解法：

 class Solution {
 public:
     int trap(vector<int>& height) {
         //小于等于两个没办法接水
         if(height.size()<=2)return 0;
         //存放下标，计算的时候使用下标对应的柱子高度
         stack<int>st;
         //存放第一个柱子下标
         st.push(0);
         //定义和
         int sum=0;
         //遍历柱子，解决面对的三种情况
         for(int i=1;i<height.size();i++){
             //情况一（高度小于栈顶高度）
             if(height[i]<height[st.top()]){
                 st.push(i);
             }
             //情况二（高度等于栈顶高度）
             if(height[i]==height[st.top()]){
                 st.pop();
                 st.push(i);
             }
             //情况三（高度大于栈顶高度）
             else{
                 while(!st.empty()&&height[i]>height[st.top()]){
                     //定义最低槽
                     int mid=st.top();
                     st.pop();
                     if(!st.empty()){
                         //找到左右最大高中较小的，计算存水高度
                         int h=min(height[st.top()],height[i])-height[mid];
                         //计算宽度
                         int w=i-st.top()-1;
                         sum+=h*w;
                     }
                 }
                 st.push(i);
             }
         }
         return sum;
     }
 };