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

 

解题参考链接：接雨水
解法一：暴力法

时间复杂度：O(n2) ，空间复杂度 O(1) 的额外空间。

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

解法二：动态规划

时间复杂度：O(n)。

存储最大高度数组，需要两次遍历，每次 O(n) 。
最终使用存储的数据更新 ans ，O(n)。
空间复杂度：O(n)额外空间。

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

解法三：双指针

class Solution {
public:
    int trap(vector<int>& height) {
        int len = height.size();
        if(len == 0)
            return 0;
        
        int left = 0,right = len-1,maxleft = 0,maxright = 0;
        int ans = 0;
        while(left < right){
            if(height[left] < height[right]){
                //增加ans时保证了当前height[left]<height[right]并且maxleft>height[left]
                maxleft<=height[left] ? maxleft=height[left]:ans+=maxleft-height[left];
                left++;
            }else{
                maxright<=height[right] ? maxright=height[right]:ans+=maxright-height[right];
                right--;
            }
        }
        return ans;
    }
};