Leetcode热题 42. 接雨水

知识点：动态规划、双指针
时间：2021年4月17日
题目链接

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

解题思路

1. 数组中的每个元素，分别向左和向右扫描并记录左边和右边的最大高度，取两边最低的高度 - 元素高度 得到每个下标位置能接的雨水量
2. 可以用left、right数组 记录左右两边最大高度 方便后面直接取
3. 可以用双指针代替数组 边走边移动

代码

#include "cheader.h"
class Solution {
public:
    int trap(vector<int>& height) {
        if(height.size() == 0)
            return 0;
        int n = height.size();
        int ans = 0;
        vector<int> left(n,0);
        vector<int> right(n,0);
        left[0] = height[0];
        right[n-1] = height[n-1];
        for(int i = 1 ; i < n ;i++)
            left[i] = max(left[i-1],height[i]);
        for(int i = n-2; i >= 0; i--)
            right[i] = max(right[i+1],height[i]);
        for(int i = 0; i < n; i++){
            ans += min(left[i],right[i])-height[i];
        }
        return ans;
    }
};

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

今天也是爱zz的一天哦！