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。