LeetCode-42-接雨水

题目

<困难> 接雨水
来源：LeetCode.

给定 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$
• $1<=n<=2\ast 104$
• $0<=height[i]<=105$

接下来看一下解题思路：

思路一：双指针：

    下标 $i$ 处能接的雨水量由 $\text{leftMax}[i]$ 和 $\text{rightMax}[i]$ 中的最小值决定。由于数组 $\text{leftMax}$ 是从左往右计算，数组 $\text{rightMax}$ 是从右往左计算，因此可以使用双指针。

    维护两个指针 $\text{left}$ 和 $\text{right}$，以及两个变量 $\text{leftMax}$ 和 $\text{rightMax}$，初始时 $\text{left}=0,\text{right}=n-1,\text{leftMax}=0,\text{rightMax}=0$。指针 $\text{left}$ 只会向右移动，指针 $\text{right}$ 只会向左移动，在移动指针的过程中维护两个变量 $\text{leftMax}$ 和 $\text{rightMax}$ 的值。

    当两个指针没有相遇时，进行如下操作：

• 使用 $\text{height}[\text{left}]$ 和 $\text{height}[\text{right}]$ 的值更新 $\text{leftMax}$ 和 $\text{rightMax}$ 的值；

• 如果 $\text{height}[\text{left}]<\text{height}[\text{right}]$，则必有 $\text{leftMax}<\text{rightMax}$，下标 $\text{left}$ 处能接的雨水量等于 $\text{leftMax}-\text{height}[\text{left}]$，将下标 $\text{left}$ 处能接的雨水量加到能接的雨水总量，然后将 $\text{left}$ 加 $1$（即向右移动一位）；

• 如果 $\text{height}[\text{left}]\ge \text{height}[\text{right}]$，则必有 $\text{leftMax}\ge \text{rightMax}$，下标 $\text{right}$ 处能接的雨水量等于 $\text{rightMax}-\text{height}[\text{right}]$，将下标 $\text{right}$ 处能接的雨水量加到能接的雨水总量，然后将 $\text{right}$ 减 $1$（即向左移动一位）。

当两个指针相遇时，即可得到能接的雨水总量。

class Solution {
    public int trap(int[] height) {
        int n = height.length;
        if (n < 3) {
            return 0;
        }

        int left = 0;
        int right = n - 1;
        int leftMax = 0;
        int rightMax = 0;
        int res = 0;

        while (left < right){
            leftMax = Math.max(leftMax, height[left]);
            rightMax = Math.max(rightMax, height[right]);

            if (leftMax < rightMax) {
                res += (leftMax - height[left]);
                ++left;
            } else {
                res += (rightMax - height[right]);
                --right;
            }
        }
        return res;
    }
}

复杂度分析

• 时间复杂度：$O(n)$，其中 $n$ 是数组 $\text{height}$ 的长度。两个指针的移动总次数不超过 $n$。
• 空间复杂度：$O(1)$。只需要使用常数的额外空间。