【力扣】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$

提示

1. $n==height.length$
2. $1<=n<=2\ast 10^4$
3. $0<=height[i]<=10^5$

官方题解

双指针

算法总结

1. 初始化两个指针 $left=0$ 和 $right=height.length-1$。
2. 当 $left<right$ 时向中间移动两个指针：
   2.1 如果 $height[left]<height[right]$，说明储水量依赖于 $height[left]$ 的高度（可能构成低洼的右边界很大）
   2.1.1 如果 $height[left]>leftMax$ 说明没有或超出左边边界，不构成低洼，$leftMax=height[left]$。
   2.1.2 如果 $height[left]<=leftMax$ 说明构成低洼，往答案中累加积水量。$ans+=leftMax-height[left]$。
   2.1.3 前进 $left$，$left++$。
   2.2 如果 $height[left]>=height[right]$，说明储水量依赖于 $height[right]$ 的高度（可能构成低洼的左边界很大）
   2.2.1 如果 $height[right]>rightMax$ 说明没有或超出右边边界，不构成低洼，$rightMax=height[right]$。
   2.2.2 如果 $height[right]<=rightMax$ 说明构成低洼，往答案中累加积水量。$ans+=rightMax-height[right]$。
   2.2.3 前进 $right$，$right--$。

复杂度

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

本人代码

Java

class Solution {
    public int trap(int[] height) {
        int left = 0, right = height.length - 1;
        int leftMax = 0, rightMax = 0;
        int ans = 0;

        while (left < right) {
            if (height[left] < height[right]) {
                if (height[left] >= leftMax) {
                    leftMax = height[left];
                } else {
                    ans += leftMax - height[left];
                }
                left++;
            } else {
                if (height[right] >= rightMax) {
                    rightMax = height[right];
                } else {
                    ans += rightMax - height[right];
                }
                right--;
            }
        }

        return ans;
    }
}

提交结果：通过

• 执行用时: $1ms$
• 内存消耗: $42.42MB$