Leetcode Trapping Rain Water

最新推荐文章于 2025-12-16 18:10:48 发布

原创最新推荐文章于 2025-12-16 18:10:48 发布 · 294 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

167 篇文章

订阅专栏

本文介绍了一种使用双数组记录左右两侧最大高度的方法来解决计算地形中能汇集的最大雨水量问题。通过一次遍历，该算法可以高效地找到每个位置上的积水高度。

题意：给出地形的高度，求出能汇集的最多的雨水。

思路：这题和什么时候买股票那题有点像。数组lr记录到目前为止，左边的最高高度；数组rl记录到目前为止，右边的最高高度。他们的最小值就是能汇集的水量。

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size() <= 1) return 0;
        vector<int> lr;
        vector<int> rl;
        int maxf = height[0];
        for(int i = 0; i < height.size(); ++ i) {
            if(maxf < height[i]) maxf = height[i];
            lr.push_back(maxf);
        }
        maxf = height[height.size() - 1];
        for(int i = height.size() - 1; i >= 0; i --) {
            if(maxf < height[i]) maxf = height[i];
            rl.push_back(maxf);
        }
        std::reverse(rl.begin(), rl.end());
        
        int water = 0;
        int top = 0;
        for(int i = 0; i < height.size(); ++ i) {
            top = min(lr[i], rl[i]); //cout << top <<" " << lr[i] << " " << rl[i] << endl;
            water += (top - height[i]);
        }
        return water;
    }
};