力扣刷题打卡day4 接雨水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 个单位的雨水（蓝色部分表示雨水）。

思路一：暴力解法

• 从下至上依次计算”每一层存水量“
• 时间复杂度为$O(N^2)$
• 毫无疑问的超时

class Solution {
public:
    int trap(vector<int>& height) {
        int res=0;
        for(int floor=0,left=-1;floor<*max_element(height.begin(),height.end());){
            for(int pos=0;pos<height.size();pos++){
                if(height[pos]>0){
                    if(left==-1){
                        left=pos;
                    }else{
                        res+=pos-left-1;
                        left=pos;
                    }
                }
                height[pos]--;
            }
            left=-1;
        }
        return res;
    }
};

思路二：动态规划（力扣官解）

• 遍历height，计算出每个位置左方和右方的最大高度值leftmax与rightmax，再依次计算min(leftmax[i],rightmax[i])-height[i]，求和得到最后结果

int trap(vector<int>& height) {
    const int len=10000;
    int leftmax[len],rightmax[len],res=0;
    int left=height[0],right=height[height.size()-1];
    for(int i=0;i<height.size();i++){
        //left
        leftmax[i]=max(height[i],left);
        left=leftmax[i];
        //right
        rightmax[height.size()-1-i]=max(height[height.size()-1-i],right);
        right=rightmax[height.size()-1-i];
    }
    for(int i=0;i<height.size();i++){
        res+=min(leftmax[i],rightmax[i])-height[i];
    }
    return res;
}