【LeetCode】42. 接雨水(困难)——代码随想录算法训练营Day59

题目链接:42. 接雨水

题目描述

给定 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 * 104
  • 0 <= height[i] <= 105

文章讲解:代码随想录

视频讲解:单调栈,经典来袭!LeetCode:42.接雨水_哔哩哔哩_bilibili

题解1:暴力法

思路:纵向求解。先找出每个柱子左边和右边的最高柱子,当前柱子能接的水量为左右最高柱子小的一方减去当前柱子的高度。

javascript 复制代码
/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function(height) {
    const left = [height[0]];
    const right = [];
    right[height.length - 1] = height[height.length - 1];
    for (let i = 1; i < height.length; i++) {
        left[i] = Math.max(left[i - 1], height[i]);
    }
    for (let i = height.length - 2; i>= 0; i--) {
        right[i] = Math.max(right[i + 1], height[i]);
    }
    let res = 0;
    for (let i = 1; i < height.length - 1; i++) {
        res += Math.min(left[i], right[i]) - height[i];
    }
    return res;
};

分析:时间复杂度为 O(n),空间复杂度为 O(n)。

题解2:单调栈

思路:横向求解。使用单调栈找到左边和右边比当前柱子高的第一个柱子。做弹出操作时,弹出的元素为当前柱子,当前遍历的元素为右边比当前柱子高的第一个柱子,栈顶元素为左边比当前柱子高的第1个柱子。能接的雨水长方形高为左右比当前柱子高的第一个柱子低的一方减去当前柱子高度,底为当前遍历的位置减去栈顶位置再减1,累加雨水得出答案。

javascript 复制代码
/**
 * @param {number[]} height
 * @return {number}
 */
var trap = function(height) {
    let res = 0;
    const stack = [];
    for (let i = 0; i < height.length; i++) {
        while (stack.length > 0 && height[stack[stack.length - 1]] < height[i]) {
            const mid = stack.pop();
            if (stack.length > 0) {
                res += (Math.min(height[stack[stack.length - 1]], height[i]) - height[mid]) * (i - stack[stack.length - 1] - 1); // 高×底
            }
        }
        if (stack.length > 0 && height[stack[stack.length - 1]] === height[i]) {
            stack.pop();
        }
        stack.push(i);
    }
    return res;
};

分析:时间复杂度为 O(n),空间复杂度为 O(n)。

收获

练习单调栈解经典题目接雨水的方法,本质是用单调栈找出当前元素左右比他大的第1个元素。

相关推荐
go546315846515 分钟前
修改Spatial-MLLM项目,使其专注于无人机航拍视频的空间理解
人工智能·算法·机器学习·架构·音视频·无人机
油泼辣子多加32 分钟前
【Torch】nn.BatchNorm1d算法详解
算法
nlog3n33 分钟前
基于 govaluate 的监控系统中,如何设计灵活可扩展的自定义表达式函数体系
算法·go
IT古董1 小时前
【第三章:神经网络原理详解与Pytorch入门】01.神经网络算法理论详解与实践-(2)神经网络整体结构
pytorch·神经网络·算法
ThetaarSofVenice1 小时前
垃圾收集相关算法Test
java·jvm·算法
小陈phd1 小时前
langchain从入门到精通(二十八)——RAG优化策略(六)集成多种检索器算法实现混合检索及问题转换总结
算法
是小王同学啊~1 小时前
(LangChain)RAG系统链路向量检索器之Retrievers(五)
python·算法·langchain
薰衣草23332 小时前
一天两道力扣(1)
算法·leetcode·职场和发展
一粒沙白猫2 小时前
Java综合练习04
java·开发语言·算法
爱coding的橙子2 小时前
每日算法刷题Day41 6.28:leetcode前缀和2道题,用时1h20min(要加快)
算法·leetcode·职场和发展