【LeetCode算法学习笔记】接雨水问题

labuladong公众号算法学习笔记

就是用一个数组表示一个条形图，问这个条形图最多能接多少雨水。

暴力解法

核心思路：
位置i能达到的水柱高度和其左边的最高柱字、右边的最高柱字有关。位置i的最大水柱高度为min(l_max，r_max)。

water[i] = min( //左边最高的柱子
                max(height[0...i],
                //右边最高的柱子
                max(height[i...end]
               ) - height[i]

暴力算法：

int trap(vector<int>& height){
    int n = height.size();
    int res = 0;
    for(int i = 1; i < n-1; i++){
        int l_max = 0, r_max = 0;
        //寻找右边最高的柱子
        for(int j = i;j < n; j++){
            r_max = max(r_max,height[j]);
        }
        //寻找左边最高的柱子 
        for(int j = i;j >= 0; j--){
            l_max = max(l_max,height[j]);
        }
        //如果自己就是最高的话
        //l_max = r_max = height[i]
        res += min(l_max,r_max) - height[i];
    }
    return res;
}

备忘录优化

开两个数组r_max和l_max充当备忘录，l_max[i]表示位置i左边的最高柱子高度，r_max[i]表示位置i右边的最高柱子高度。
可以先把这两个数组计算好，避免重复计算。

int trap(vector<int>& height){
    if(height.empty()) return 0;
    int n = height.size();
    int res = 0;
    //数组充当备忘录
    vector<int> l_max(n), r_max(n);
    //初始化
    l_max[0] = height[0];
    r_max[n-1] = height[n-1];
    //从左向右计算l_max
    for(int i = 1;i < n; i++) l_max[i] = max(height[i],l_max[i-1]);
    //从右向左计算r_max
    for(int i = n-2; i >= 0; i--) r_max[i] = max(height[i],r_max[i+1]);
    //计算答案
    for(int i = 1; i < n-1;i++) res += min(l_max[i],r_max[i]) - height[i];
    return res;
}

双指针解法

双指针边走边算，节省下空间复杂度。

int trap(vector<int>& height){
    if(height.empty()) return 0;
    int n = height.size();
    int left = 0, right = 0;
    int res = 0;
    
    int l_max = height[0];
    int r_max = height[n-1];
    
    while(left < right){
        l_max = max(l_max,height[left]);
        r_max = max(r_max,height[right]);
        
        //res += min(l_max,r_max)-height[i];
        if(l_max < r_max){
            res += l_max - height[left]; 
            left++;
        }else{
            res += r_max - height[right];
            right--;
        }
    }
    return res;
}

之前备忘录的解法，l_max和r_max代表的是height[0…i]和height[i…n-1]的最高柱子高度。

但是双指针的解法中，l_max和r_max代表的是height[0…left]和height[right…n-1]的最高柱子高度。（此时l_max是left指针左边的最高柱子，但是r_max不一定是left指针右边的最高柱子。）