Leetcode Trapping Rain Water

markpen

于 2016-12-22 05:44:52 发布

阅读量281

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 文章标签： leetcode

本文链接：https://blog.youkuaiyun.com/markpen/article/details/53808241

leetcode 专栏收录该内容

167 篇文章

订阅专栏

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

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

思路：这题和什么时候买股票那题有点像。数组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;
    }
};