双指针解决“接雨水”

 题目描述

给定 n 个非负整数表示每个宽度为 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 个单位的雨水（蓝色部分表示雨水）。 

提示：

• n == height.length

• 1 <= n <= 2 * 104

• 0 <= height[i] <= 105

解题思路

这个问题可以通过双指针的方法来解决。我们使用两个指针 left 和 right，分别从数组的两端开始遍历，计算每个位置能接住的雨水量。

算法步骤：

1. 初始化：定义两个指针 left 和 right，分别指向数组的开始和结束位置。同时定义两个变量 leftMax 和 rightMax 来记录从左到右和从右到左遍历过程中遇到的最大高度。
2. 遍历：从左到右和从右到左同时遍历数组。
   • 对于从左到右的遍历，如果当前高度小于 leftMax，则当前位置的雨水量为 leftMax - height[i]，否则更新 leftMax 为当前高度。
   • 对于从右到左的遍历，如果当前高度小于 rightMax，则当前位置的雨水量为 rightMax - height[i]，否则更新 rightMax 为当前高度。
3. 累加雨水量：将每次遍历中计算的雨水量累加到总雨水量中。
4. 返回结果：返回总雨水量。

代码实现

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

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

        return water;
    }
}

复杂度分析

• 时间复杂度：O(n)。只需要遍历数组一次。
• 空间复杂度：O(1)。只使用了常数级别的额外空间。

解释

1. 初始化：设置两个指针 left 和 right，以及两个用于记录最大高度的变量 leftMax 和 rightMax。
2. 遍历：通过比较 height[left] 和 height[right] 来决定是向左移动还是向右移动，并更新 leftMax 或 rightMax。
3. 计算雨水量：如果当前高度小于对应的最大高度，则计算当前位置的雨水量，并累加到总雨水量中。
4. 返回结果：返回计算得到的总雨水量。