面试题 17.21. 直方图的水量--leecode

题目

给定一个直方图(也称柱状图)，假设有人从上面源源不断地倒水，最后直方图能存多少水量?直方图的宽度为 1。
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
题目

解题思路

ldp[i]是从左边开始遍历。
ldp[i]表示第i个位置左边最高的边界。
rdp[i]是从右边开始遍历。
rdp[i]表示第i个位置右边最高的边界。
第i个位置的存水量=min(ldp[i],rdp[i]) - height[i];
对于题目中给的测试用例，两个数组的结果如下：
0 (0,3)
1 (1,3)
2 (1,3)
3 (2,3)
4 (2,3)
5 (2,3)
6 (2,3)
7 (3,3)
8 (3,2)
9 (3,2)
10 (3,2)
11 (3,1)
代码如下：

class Solution {
public:
    int trap(vector<int>& height) {
        int l = height.size();
        if(l == 0) return 0;
        vector<int> lDp(l,0);
        vector<int> rDp(l,0);
        lDp[0] = height[0], rDp[l - 1] = height[l - 1]; 
        for(int i = 1;i < l;i++){
            lDp[i] = max(lDp[i - 1],height[i]);
            rDp[l - i - 1] = max(rDp[l - i],height[l - i - 1]);
        }
        int ans = 0;
        for(int i = 0;i < l;i++){
            ans += min(lDp[i],rDp[i]) - height[i];
        }
        return ans;
    }
};