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

思路：一个柱子能接多少水，取决于他前面的全部柱子和后面的全部柱子的最小值。

也就是说：取决于当前柱子的前缀和后缀。这样就可以抽象成为板子，能接的水就等于当前柱子的前缀后缀最小值减去柱子高度。

所以我们需要定义两个数组，来记录每个柱子的前后缀。

代码如下：时间复杂度o(3n)=o(n);空间复杂度o(2n)=o(n);

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

优化:
对于第一个元素/最后一个元素来说，前缀/后缀是确定的。
第一个元素:前缀等于height[0]这是已知的;虽然不知道后缀，但是只要满足当前元素的前缀比最后已知元素的后缀小，那么第一个柱子前缀一定比后面的后缀小!
pre_max<suf_max(当前元素的最大前缀比已知元素的后缀小，那么当前元素的最大前缀一定小于最大后缀)
所以当前第一个柱子的接水量:pre_max-height[0];
同样，对于最后一个元素而言，它的最大后缀是知道的
如果suf_max<pre_max，则接水量: suf_max-height[n-1];
所以利用第一个元素和最后一个元素的特性，使用首尾指针进行逐步移动求解。

代码如下:可以看出，时间复杂度依旧是o(n)，但空间复杂度为o(1）。

class Solution {
public:
    int trap(vector<int>& height) {
        int left=0;
        int right=height.size()-1;
        int pre_max=0;
        int suf_max=0;
        int ans=0;
        while(left<=right){
            pre_max=max(pre_max,height[left]);
            suf_max=max(suf_max,height[right]);
            if(pre_max<suf_max){
                ans+=pre_max-height[left];
                left++;
            }  
            else{
                ans+=suf_max-height[right];
                right--;
            }               
        }
        return ans;
    }
};