LeetCode 力扣 热题 100道（二十一）接雨水（C++）

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

代码如下所示：

class Solution {
public:
    int trap(vector<int>& height) {
        int n = height.size();
        if (n == 0) return 0;

        int left = 0, right = n - 1;
        int left_max = 0, right_max = 0;
        int water = 0;

        while (left < right) {
            if (height[left] < height[right]) {
                if (height[left] >= left_max) {
                    left_max = height[left]; // 更新左侧最大高度
                } else {
                    water += left_max - height[left]; // 累加雨水
                }
                ++left; // 左指针右移
            } else {
                if (height[right] >= right_max) {
                    right_max = height[right]; // 更新右侧最大高度
                } else {
                    water += right_max - height[right]; // 累加雨水
                }
                --right; // 右指针左移
            }
        }

        return water;
    }
};

利用两个指针分别从数组的两端向中间遍历。

同时维护两个变量：

left_max：从左侧当前最大高度。

right_max：从右侧当前最大高度。

计算能接的雨水：

如果 height[left] 小于等于 height[right]，则 left 所在位置的最大雨水由 left_max - height[left] 决定，更新左指针。

如果 height[right] 小于 height[left]，则 right 所在位置的最大雨水由 right_max - height[right] 决定，更新右指针。

继续移动指针，直到 left == right。