LeetCode 42 接雨水

给定 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##

思路

首先我们遍历一次数组，找到数组中最大的元素位置maxInd，也就是最高的柱子，然后分别从数组的两端向maxInd方向遍历。因为我们找到了最大的元素maxlnd，所在在从一端向maxInd方向遍历的时候就不用担心另一端会泄露（柱子矮），这样做，我们就可以一边遍历一边增加水量area。不说了，直接上代码

C++ 代码

class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size() <= 2)
            return 0;
        int i = 0;
        int max = -1;
        int maxInd = 0;
        int area = 0;
        for(; i < height.size(); i++) {
            if(height[i] > max) {
                max = height[i];
                maxInd = i;
            }
        }
        int maxLeft = -1;//存放目前左边的最大值
        for(i = 0; i < maxInd; i++) {
            if(height[i] > maxLeft)
                maxLeft = height[i];
            else
                area = area + (maxLeft - height[i]);
        }
        int maxRight = -1; //存放目前右边的最大值
        for(i = height.size() - 1; i > maxInd; i--) {
            if(height[i] > maxRight)
                maxRight = height[i];
            else
                area = area + (maxRight - height[i]);
        }
        return area;
    }
};