42. [C++]接雨水

题目描述

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：
输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。

示例 2：
输入：height = [4,2,0,3,2,5]
输出：9

提示：

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

解题思路

i处雨水高度=i处左右最高高度的最小值-i处高度

代码

class Solution {
public:
    int trap(vector<int>& height) {
        int n=height.size();
        int maxLeft[n];//记录第i个位置上左边的最大高度
        int maxRight[n];//记录第i个位置上右边的最大高度
        int maxHeight[n];//记录第i个位置上的最大高度
        int sum=0;
        int i;
        //正向遍历，找出每个位置左边的最大高度
        maxLeft[0]=height[0];
        for(i=1;i<n;i++){
            maxLeft[i]=height[i]>maxLeft[i-1]?height[i]:maxLeft[i-1];
        }
        //反向遍历，找出每个位置右边的最大高度
        maxRight[n-1]=height[n-1];
        for(i=n-2;i>=0;i--){
            maxRight[i]=height[i]>maxRight[i+1]?height[i]:maxRight[i+1];
        }
        //找出每个位置的
        for(i=0;i<n;i++){
            maxHeight[i]=maxLeft[i]<maxRight[i]?maxLeft[i]:maxRight[i];
            sum+=(maxHeight[i]-height[i]);
        }
        return sum;
        
    }
};