42. 接雨水

https://leetcode-cn.com/problems/trapping-rain-water/

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

 

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

 

// 思路：
// 1. 先找出最高处的山峰
// 2. 从左边一直搜到最高峰，记录左边高度，若发现左边大于右边，则说明可以装水
// 3. 从右边一直搜到最高峰，若发现右边大于左边，则说明可以装水


class Solution {
public:
    int trap(vector<int>& height) {
        int max_index = 0;
        for (int i = 1; i < height.size(); ++i) {
            if (height[i] > height[max_index]) {
                max_index = i;
            }
        }
        
        int left_height = 0, right_height = 0;
        int area = 0;
        for (int i = 0; i < max_index; ++i) {
            if (height[i] >= left_height) {
                left_height = height[i];
            } else {
                area += left_height - height[i];
            }
        }
        
        for (int i = height.size() - 1; i > max_index; --i) {
            if (height[i] >= right_height) {
                right_height = height[i];
            } else {
                area += right_height - height[i];
            }
        }
        
        return area;
    }
};