LeetCode 力扣 热题 100道(二十一)接雨水(C++)

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

代码如下所示:

cpp 复制代码
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

相关推荐
Theodore_102211 分钟前
深度学习(3)神经网络
人工智能·深度学习·神经网络·算法·机器学习·计算机视觉
nexttake14 分钟前
5.go-zero集成gorm 和 go-redis
开发语言·后端·golang
地平线开发者25 分钟前
大模型 | VLA 初识及在自动驾驶场景中的应用
算法·自动驾驶
py有趣27 分钟前
LeetCode学习之实现strStr()
学习·算法·leetcode
夏鹏今天学习了吗30 分钟前
【LeetCode热题100(52/100)】课程表
算法·leetcode·职场和发展
祁思妙想30 分钟前
【LeetCode100】--- 101.重排链表【思维导图+复习回顾】
算法·leetcode·链表
VBA633741 分钟前
VBA技术资料MF383:处理Excel中存储为文本的数据
开发语言
Dream it possible!43 分钟前
LeetCode 面试经典 150_链表_随机链表的复制(59_138_C++_中等)
c++·leetcode·链表
CiLerLinux1 小时前
第三章 FreeRTOS 任务相关 API 函数
开发语言·单片机·物联网·c#
我是华为OD~HR~栗栗呀1 小时前
华为od-22届考研-C++面经
java·前端·c++·python·华为od·华为·面试