Leetcode——42. Trapping Rain Water(C++)

最新推荐文章于 2024-07-18 23:31:00 发布

begforgood

最新推荐文章于 2024-07-18 23:31:00 发布

阅读量176

点赞数

分类专栏： Leetcode 笔记文章标签： c++ leetcode 算法

本文链接：https://blog.csdn.net/begforgood/article/details/125792519

版权

Leetcode 同时被 2 个专栏收录

11 篇文章 0 订阅

订阅专栏

笔记

10 篇文章 0 订阅

订阅专栏

Leetcode——42. Trapping Rain Water(C++)
Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it can trap after raining.

Example 1:
在这里插入图片描述 Input: height = [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6
Explanation: The above elevation map (black section) is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped.

Example 2:
Input: height = [4,2,0,3,2,5]
Output: 9

Constraints:
n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105

解题思路：记录每点左边最高值和右边最高值，二者较小者减去此点高度即是此点可蓄水值，此点高于左右最高值则此点不蓄水；

代码；

class Solution {
public:
    int trap(vector<int>& height) {
        int siz=height.size();
        vector<int>leftmax(siz,0),rightmax(siz,0);
        int lem=0,rma=0;
        for(int i=0;i<siz;i++)
        {
            leftmax[i]=lem;
            if(height[i]>lem)lem=height[i];
        }
        for(int j=siz-1;j>=0;j--)
        {
            rightmax[j]=rma;
            if(height[j]>rma)rma=height[j];
        }
        int sum=0;
        for(int i=1;i<siz-1;i++)
        {
            if(height[i]<min(leftmax[i],rightmax[i]))
                sum+=min(leftmax[i],rightmax[i])-height[i];
        }
        return sum;
    }
};