Leetcode——42. 接雨水

还记得第一次见该题根本无从下手。其实,我们不妨把问题拆解,简单化。不要怕自己写的是暴力算法,有很多算法技巧其实就是在暴力算法的基础上优化得来。

题目目的是求所有可接雨水数量,我们可以求出每一个位置可接雨水数量,最后加起来即可。

就是如下流程:

cpp 复制代码
int trap(vector<int>& height) {
        int n = height.size();
        vector<int> water(n, -1);
        for(int i = 0; i < n; ++i){
            // 求height[i]能存多少水
            water[i] = getWater(height, i);
        }
        int ans = 0;
        for(int x : water){
            ans += x;
        }
        return ans;
    }

那么现在问题只有一个,如何求单个位置的可接雨水量,根据题目,不难想到,只需要找左右两边可以"望"到的最高值,选取二者最小值,减去该位置高度即可。

完整代码:

cpp 复制代码
class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        vector<int> water(n, -1);
        for(int i = 0; i < n; ++i){
            // 求height[i]能存多少水
            water[i] = getWater(height, i);
        }
        int ans = 0;
        for(int x : water){
            ans += x;
        }
        return ans;
    }

    int getWater(vector<int>& height, int k){
        int n = height.size();
        // 计算左高点
        int lmax = height[k];
        for(int i = k - 1; i >= 0; --i){
            if(height[i] > lmax){
                lmax = height[i];
            }
        }
        // 计算右高点
        int rmax = height[k];
        for(int i = k + 1; i < n; ++i){
            if(height[i] > rmax){
                rmax = height[i];
            }
        }

        // 返回结果
        return min(lmax, rmax) - height[k];
    }
};

不过这肯定是超时的。

在此基础上,分析算法中重复计算的部分,我们在每一个位置得到 lmaxrmax 时都从零开始计算,这样很浪费算力。由于求 lmaxrmax 本质就是简单的求单调最大值,所以我们可以一遍遍历求出每个位置的lmaxrmax ,因为我们只需要向对应方向"望"到的最大值即可。

cpp 复制代码
class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        vector<int> lmax(n, 0);
        vector<int> rmax(n, 0);
        lmax[0] = height[0];
        for(int i = 1; i < n; ++i){
            lmax[i] = max(lmax[i - 1], height[i]);
        }
        rmax[n - 1] = height[n - 1];
        for(int i = n - 2; i >= 0; --i){
            rmax[i] = max(rmax[i + 1], height[i]);
        }
        int ans = 0;
        for(int i = 0; i < n; i++){
            ans += min(lmax[i], rmax[i]) - height[i];
        }
        return ans;
    }
};

最后,考虑是否可以继续优化时间和空间。

我们假设只有两个变量,分别记录 1 位置的 lmaxn - 2 位置的 rmax ,考虑现在谁的雨水量是可求的。

思考分析后得出,当两个变量进行比较,较小的一方所指位置的雨水量是可求的。

lmax 指向 1 位置,值为100rmax 指向 n - 2 为位置,值为70 为例,即使当前 lmax 对于 n - 2 位置来说并不是真正的 lmax ,但其真正值只会比100大,而接雨水量是由小的一方决定的。

cpp 复制代码
class Solution {
public:
    int trap(vector<int>& height) {
        int lidx = 1;
        int lmax = height[0];
        int ridx = height.size() - 2;
        int rmax = height[ridx + 1];
        int ans = 0;
        while(lidx <= ridx){
            if(lmax > rmax){
                ans += max(0, rmax - height[ridx]);
                rmax = max(rmax, height[ridx--]);
            }else{
                ans += max(0, lmax - height[lidx]);
                lmax = max(lmax, height[lidx++]);
            }
        }
        return ans;
    }
};
相关推荐
CoovallyAIHub5 分钟前
工业质检新突破!YOLO-pdd多尺度PCB缺陷检测算法实现99%高精度
深度学习·算法·计算机视觉
gb42152875 分钟前
负载均衡算法中的加权随机算法
windows·算法·负载均衡
xdlka1 小时前
C++初学者4——标准数据类型
开发语言·c++·算法
go54631584651 小时前
大规模矩阵构建与高级算法应用
线性代数·算法·矩阵
向左转, 向右走ˉ2 小时前
为什么分类任务偏爱交叉熵?MSE 为何折戟?
人工智能·深度学习·算法·机器学习·分类·数据挖掘
云边有个稻草人2 小时前
【C++】第十九节—一文万字详解 | AVL树实现
数据结构·c++·avl树·avl树的插入·avl树的旋转·avl树实现·avl树的结构
霜绛3 小时前
机器学习笔记(四)——聚类算法KNN、Kmeans、Dbscan
笔记·算法·机器学习·kmeans·聚类
晨非辰3 小时前
#C语言——学习攻略:深挖指针路线(三)--数组与指针的结合、冒泡排序
c语言·开发语言·数据结构·学习·算法·排序算法·visual studio
zzywxc7874 小时前
编程算法在金融、医疗、教育、制造业等领域的落地案例
人工智能·算法·金融·自动化·copilot·ai编程
zzywxc7874 小时前
编程算法在金融、医疗、教育、制造业的落地应用。
人工智能·深度学习·算法·机器学习·金融·架构·开源