LeetCode #42 接雨水

42. 接雨水

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

没搞懂这题为啥题解给出的O(n2)可以过呀，，，，，

给出一个O(n)的算法吧，只不过O(n)的有点多

也怪自己很久没打比赛，很多边界样例都考虑的不全。。。

思路：

预处理记录当前位置后面的最大值(因为如果存在比当前值大的数，后面是可以存雨水的)，然后遍历计算就行了。可以省掉一层for循环，虽然我也用了两层循环，但i的变化是根据j来的，总共也就遍历了一次。

如果没有存在比当前值大的数，也不能说明后面的不能盛雨水，如5 0 4，这时候就需要反向再遍历一次，只需要遍历后面的部分就可以了

 

其实这个方法优化的空间很大，例如记录前缀和，就不要遍历求和了、记录当前位置后面比他大的第一个位置、for循环的第一层i也可以去掉的。。。。人太懒了 不优化了

class Solution {
public:
    int out = 0;
    int F(vector<int> height){
        if(height.size()<=1) return NULL;
        int maxn[50005]={0};
        int temp = -1;
        height.push_back(0);
        for(int i = height.size();i > 0;i--){
            maxn[i-1] = max(temp,height[i]); 
            temp = maxn[i-1];
        }
        for(int i = 0;i < height.size();){
            if(maxn[i] >= height[i]){
                int temp = 0;
                for(int j = i + 1;j < height.size();j++){
                    if(height[i]>height[j]){
                        temp += height[j];
                    }
                    else{
                        out += (j-i-1)*height[i];
                        out -= temp;
                        i = j;
                        break;
                    }
                }
            }
            else return i;
        }
        return -1;
    }
    int trap(vector<int>& height) {
        if(height.size()<=1) return NULL;
        vector<int> fz = height;
        int t1 = F(height);
        if(t1 == -1) return out;
        vector<int> Height;
        for(int i = fz.size()-1;i >= t1;i--) Height.push_back(fz[i]);
        int t2 = F(Height);
        return out;
    }
};