LC 面试题 17.21. 直方图的水量

题目连接

将可以蕴含的总水量划分成局部区间来求

当前区间可以蕴含的水量 = min(其左边区间最大高度, 其右边区间的最大高度) - 当前区间的高度

LC特色：特判一下（n == 0时），不然要RE

C++ 代码

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        if (n == 0) return 0;
        vector<int> left(n), right(n);
        left[0] = height[0], right[n - 1] = height[n - 1];
        for (int i = 1; i < n; i ++ ) 
            left[i] = max(left[i - 1], height[i]);
        for (int i = n - 2; i >= 0; i -- ) 
            right[i] = max(right[i  + 1], height[i]);
        int res = 0;
        for (int i = 0 ; i < n; i ++ ) 
            res += min(left[i], right[i]) - height[i];
        return res;
    }
};