LeetCode 42. 接雨水(C语言详解)——双指针经典解法

一、题目描述

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

示例 1:

复制代码
输入:
height = [0,1,0,2,1,0,1,3,2,1,2,1]

输出:
6

解释:

该高度图在下雨后可以接到 6 个单位的雨水

示例 2:

复制代码
输入:
height = [4,2,0,3,2,5]
输出:
9

二、问题分析

每个位置能接多少雨水,取决于:

复制代码
min(左侧最高柱子, 右侧最高柱子) - 当前柱子高度

例如:

复制代码
      █
  █   █
█ █ █ █

如果某一位置:

复制代码
左边最高 = 4
右边最高 = 3
当前高度 = 1

那么该位置可以接的水为:

复制代码
min(4,3) - 1 = 2

三、双指针解法(最优解)

为了高效计算,我们可以使用 双指针 + 动态维护左右最大值

定义两个指针:

复制代码
left  = 0
right = n - 1

同时维护:

复制代码
leftMax  = 左侧最大高度
rightMax = 右侧最大高度

核心思想:

  • 如果 height[left] < height[right]

    • 说明左侧是短板

    • leftMax 计算水量

  • 否则

    • 右侧是短板

    • rightMax 计算水量

因为水位始终取决于 较矮的一侧


四、算法流程

  1. 初始化指针

    left = 0
    right = n - 1

  2. 记录左右最大高度

    leftMax = 0
    rightMax = 0

  3. 遍历数组

    while(left < right)

  4. 判断左右高度关系

若:

复制代码
height[left] < height[right]

更新左侧:

复制代码
leftMax = max(leftMax, height[left])

water += leftMax - height[left]

否则更新右侧:

复制代码
rightMax = max(rightMax, height[right])

water += rightMax - height[right]
  1. 移动指针

直到 left >= right


五、C语言代码实现

复制代码
int trap(int* height, int heightSize) {

    int left = 0;
    int right = heightSize - 1;

    int leftMax = 0;
    int 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)

只遍历一次数组,并且没有使用额外数组空间。


七、示例分析

输入:

复制代码
height = [0,1,0,2,1,0,1,3,2,1,2,1]

计算过程会累计得到:

复制代码
1 + 1 + 2 + 1 + 1 = 6

最终结果:

复制代码
6

八、总结

本题是 经典的双指针问题,核心思想是:

复制代码
水位 = min(左侧最高柱子, 右侧最高柱子)

通过双指针不断向中间收缩,并动态维护 leftMaxrightMax,就可以在 O(n) 时间内计算出所有雨水量。

这种方法相比使用 leftMax[]rightMax[] 数组的解法,空间复杂度从 O(n) 降到了 O(1),是面试中更推荐的写法。

相关推荐
「QT(C++)开发工程师」18 分钟前
C++ 11 常用for循环
开发语言·c++
我还记得那天28 分钟前
0 初识C++
开发语言·c++
Asize2 小时前
146. LRU 缓存
算法
Asize2 小时前
543. 二叉树的直径
算法
lemon_sjdk2 小时前
ObjectProperty
java·开发语言·算法
Zentceh2 小时前
AI-ISP在夜视机芯中的应用:从传统ISP到PixelClean全彩夜视的进化
人工智能·科技·算法·计算机视觉·车载系统·视频·智能硬件
「QT(C++)开发工程师」2 小时前
C++ std::move 详解
开发语言·c++
xier_ran3 小时前
【infra之路】AWQ 详解:激活感知权重保护,让 W4A16 量化精度超越 GPTQ
线性代数·算法·机器学习·量化·infra
亦皓ai3 小时前
AI时代后端工程(二):AI把代码写得越来越快,我却越来越不敢让它直接开工了
人工智能·算法·机器学习·搜索引擎·transformer