接雨水

接雨水

题目描述

来源：力扣LeetCode
给定 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
0 <= n <= 3 * 104
0 <= height[i] <= 105

想法

• 本题最关键的是要将长条会形成的雨水的坑槽
• 首先确定坑槽左边的位置，它通常是上一个确定的坑槽的右边位置，如果是第一个坑槽，那么他就是起始位置0。
• 然后确定坑槽右边的位置，它是>=左边元素的第一个位置。或者是<左边元素，但是后面已经没有更大的元素的第一个位置。
• 因此需要一个辅助的数组，记录当前位置之后的元素的大小。

代码

class Solution {
    vector<int> k;
    void initK(int n,vector<int>& height){
        k.resize(n,0);
        int mv = -1;
        for(int i=height.size()-1;i>=0;i--){
            mv = k[i] = max(mv,height[i]);  
        }
    }    
public:
    int trap(vector<int>& height) {
        vector<int> res;
        
        initK(height.size(),height);
        
        int p=0;
        res.push_back(0);
        for(int i=1;i<height.size();i++){
            if(height[i] >= height[p] || k[i] <= height[i]){
               res.push_back(i);
               p = i;
            }
        }
        int r_c = 0;
        for(int i=1;i<res.size();i++){
            int p = res[i-1];
            int q = res[i];
            int v = min(height[p],height[q]);
            for(int j=p+1;j<q;j++){
               r_c += v-height[j];
            }
        }
        return r_c;
    }
};

• 数组res记录着找到的坑槽的位置，对于第i（i>0）个坑槽，res[i-1]表示坑槽左边的位置，res[i]表示坑槽右边的位置。
• 根据木桶效应，坑槽两边元素的最小值决定了该坑槽所能储存的最大的雨水长度 v。
• 坑槽每一个位置能储藏的雨水是v-height，height是其该位置的元素的值。
• 若坑槽两边有一个元素的高度为0，那么v-height的值必是0。