leetcode 42.接雨水

哆啦刘小洋2024-01-28 23:36

问题1：怎么算接水量

总的接水量=第一列接水量+第二列接水量+第三列接水量+...+最后一列接水量

问题2：当前列的接水量怎么计算

当前的接水量=min(当前列左边最高的墙x1，当前列右边最高的墙x3）- 当前列x2的高度

问题2图解：

方法：预处理每一列左边最高的墙+预处理每一列右边最高的墙（记录下标）

cpp 复制代码

class Solution 
{
public:
    int trap(vector<int>& height) 
    {
        int n=height.size();
        int leftMax[n],rightMax[n];
        memset(leftMax,0,sizeof(int)*n);
        memset(rightMax,0,sizeof(int)*n);
        int lmax=height[0]; 
        int rmax=height[n-1];
        for(int i=1;i<n;i++)   //预处理,没自己写看起来有点小难懂，自己模拟写一个即可
        {
            leftMax[i]=lmax;
            lmax=max(lmax,height[i]);
            rightMax[n-i-1]=rmax;
            rmax=max(rmax,height[n-i-1]);
        }
        int sum=0;
        for(int i=1;i<n-1;i++)  //一列一列的算，为什么不算第一列和最后一列，因为最左边和最右边就是边界，相当于最后的墙
        {
            if(min(leftMax[i],rightMax[i])<=height[i])continue;
            sum+=min(leftMax[i],rightMax[i])-height[i];
        }
        return sum;
    }
};