leetcode 42.接雨水

rvlt1

已于 2023-03-22 11:56:30 修改

阅读量80

点赞数

文章标签： leetcode 算法职场和发展

于 2023-03-18 16:30:00 首次发布

本文链接：https://blog.csdn.net/skykone1/article/details/129638943

版权

该代码实现了一个C++类Solution，其主要功能是计算给定柱子高度数组中能接住的雨水量。通过维护左右两侧的最大高度，遍历数组计算每个柱子能接的雨水，并累加得到总雨水量。

摘要由CSDN通过智能技术生成

class Solution {
public:
    //每个柱子能接到的雨水=左边最大高度和右边最大高度的较小值减去当前柱子的高度
    //leftMax[i]表示i位置左边的最大高度
    //rightMax[i]表示i位置右边的最大高度
    int leftMax[20005];
    int rightMax[20005];
    int trap(vector<int>& height) {
        int sz = height.size();
        leftMax[0] = height[0];
        //从左边开始计算
        for(int i = 1; i < sz; i++){
            leftMax[i] = max(leftMax[i - 1], height[i]);
        }
        rightMax[sz - 1] = height[sz - 1];
        //从右边开始计算
        for(int i = sz - 2; i >= 0; i--){
            rightMax[i] = max(rightMax[i + 1], height[i]);
        }
        int ans = 0;
        for(int i = 0; i < sz; i++){
            ans += min(leftMax[i], rightMax[i]) - height[i];
        }
        return ans;
    }
};