【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个元素。

相关推荐
2401_841495641 小时前
【计算机视觉】基于复杂环境下的车牌识别
人工智能·python·算法·计算机视觉·去噪·车牌识别·字符识别
Jonkin-Ma1 小时前
每日算法(1)之单链表
算法
晚风残2 小时前
【C++ Primer】第六章:函数
开发语言·c++·算法·c++ primer
杨云强2 小时前
离散积分,相同表达式数组和公式
算法
地平线开发者2 小时前
征程 6 | BPU trace 简介与实操
算法·自动驾驶
满天星83035772 小时前
【C++】AVL树的模拟实现
开发语言·c++·算法·stl
Lris-KK3 小时前
力扣Hot100--94.二叉树的中序遍历、144.二叉树的前序遍历、145.二叉树的后序遍历
python·算法·leetcode
麦麦鸡腿堡3 小时前
Java的动态绑定机制(重要)
java·开发语言·算法
zy_destiny3 小时前
【工业场景】用YOLOv8实现抽烟识别
人工智能·python·算法·yolo·机器学习·计算机视觉·目标跟踪
坚持编程的菜鸟4 小时前
LeetCode每日一题——螺旋矩阵
c语言·算法·leetcode·矩阵