LeetCode 接雨水 双指针

原题链接:

力扣(LeetCode)官网 - 全球极客挚爱的技术成长平台

题面:

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

示例 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 个单位的雨水(蓝色部分表示雨水)。 

示例 2:

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

提示:

  • n == height.length
  • 1 <= n <= 2 * 10^4
  • 0 <= height[i] <= 10^5

解题思路:

由木桶理论可知某一列的储水量与他左边最高的柱子和右边最高的柱子之间的较矮者有关,为这一列高度和它的差值。在上一篇博客,我们使用了DP方法,维护了两个数组left和right,空间复杂度为O(n),但实际上我们可以使用双指针法,空间复杂度可以优化到O(1)。

设左指针left,右指针right,当left和right未相遇时,维护leftMax和rightMax,左指针left只从左往右,右指针right只从右往左,leftMax = max(leftMax, heightleft),rightMax = max(rightMax, heightright)。

当leftMax < rightMax时,我们可以计算出left位置的储水量为leftMax - heightleft,并将left右移一位;

当leftMax > rightMax时,我们可以计算出right位置的储水量为rightMax - heightright,并将right左移一位;

当leftMax == rightMax时,无论对left操作还是对right操作都是可以的。

当left和right相遇时,就可以结束计算了。

代码(CPP):

cpp 复制代码
class Solution {
public:
    /*
        双指针,将空间复杂度从O(n)优化至O(1)
    */
    int trap(vector<int>& height) {
        int n = height.size();
        int left = 0, right = n - 1;
        int leftMax = 0, rightMax = 0, ans = 0;
        while (left < right) {
            leftMax = max(leftMax, height[left]);
            rightMax = max(rightMax, height[right]);
            if (leftMax < rightMax) {
                ans += leftMax - height[left];
                left++;
            } else {
                ans += rightMax - height[right];
                right--;
            }
        }
        return ans;
    }
};
相关推荐
平生不晚18 分钟前
在 SVG 体系里画一条任意的线
前端·算法
evans在进步42 分钟前
LeetCode 198:打家劫舍——Java 动态规划详解
java·leetcode·动态规划
逆境不可逃1 小时前
LeetCode 双题:415. 字符串相加与 143. 重排链表
算法
2601_950760791 小时前
TNF-α:自身免疫疾病治疗的核心靶点与信号通路解析
人工智能·算法·蛋白
不会代码的小猴2 小时前
2. 了解Qt
开发语言·c++·笔记·qt·算法
ZC跨境爬虫2 小时前
LeetCode 119. 杨辉三角 II(原地更新优化详解 + Java Python 实现)
java·python·leetcode
比奇堡裤头村2 小时前
统计学习方法——支持向量机
算法·支持向量机·学习方法
吃着火锅x唱着歌2 小时前
LeetCode 3597.分割字符串
算法·leetcode·职场和发展
程序喵大人4 小时前
【C++进阶】STL算法与函数对象 - 09 函数对象保存状态并复用规则
开发语言·c++·算法·stl·函数对象
致Great9 小时前
DeepSeek Harness插件开发实战教程:我让它自己写了一个 arXiv 搜索插件
算法