【面试经典 150 | 数组】接雨水

2024-05-01 09:36
文章标签 数组 面试 经典 150 雨水

本文主要是介绍【面试经典 150 | 数组】接雨水,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

文章目录

  • 写在前面
  • Tag
  • 题目来源
  • 解题思路
    • 方法一:预处理
    • 方法二:单调栈
    • 方法三:双指针
  • 写在最后

写在前面

本专栏专注于分析与讲解【面试经典150】算法,两到三天更新一篇文章,欢迎催更……

专栏内容以分析题目为主,并附带一些对于本题涉及到的数据结构等内容进行回顾与总结,文章结构大致如下,部分内容会有增删:

  • Tag:介绍本题牵涉到的知识点、数据结构;
  • 题目来源:贴上题目的链接,方便大家查找题目并完成练习;
  • 题目解读:复述题目(确保自己真的理解题目意思),并强调一些题目重点信息;
  • 解题思路:介绍一些解题思路,每种解题思路包括思路讲解、实现代码以及复杂度分析;
  • 知识回忆:针对今天介绍的题目中的重点内容、数据结构进行回顾总结。

Tag

【数组】【单调栈】【双指针】


题目来源

42. 接雨水


解题思路

方法一:预处理

思路

一个朴素的做法是对于数组 height 中的每个元素,分别向左和向右扫描并记录左边的最大高度,然后计算每个下标位置能接到的雨水量。

假设数组 height 的长度为 n n n,朴素的做法需要对每个下标位置使用 O ( n ) O(n) O(n) 的时间向两边扫描并获得最大高度,总的时间复杂度为 O ( n 2 ) O(n^2) O(n2)

本题数据规模达到 1 0 5 10^5 105,时间复杂度为 O ( n 2 ) O(n^2) O(n2) 的解法一定超时。

预处理

上述做法的时间复杂度较高是因为需要对每个下标位置都向两边扫描。如果已经知道每个位置两边的最大高度,则可以在 O(n)O(n)O(n) 的时间内得到能接的雨水总量。

于是可以提前计算并记录每个位置两侧的最大高度。

具体地,维护数组 leftMaxrightMax,其中 leftMax[i] 表示下标 i 及其左边的位置中,height 的最大高度。rightMax[i] 表示下标 i 及其右边的位置中,height 的最大高度。显然有初始化 leftMax[0] = height[0]rightMax[n-1] = height[n-1],其余位置的元素计算比较简单,直接看代码。

代码

class Solution {
public:int trap(vector<int>& height) {int n = height.size();if (n == 0) {return 0;}// 更新 leftMaxvector<int> leftMax(n);leftMax[0] = height[0];for (int i = 1; i < n; ++i) {leftMax[i] = max(leftMax[i-1], height[i]);}// 更新 rightMaxvector<int> rightMax(n);rightMax[n-1] = height[n-1];for (int i = n - 2; i >= 0; --i) {rightMax[i] = max(rightMax[i+1], height[i]);}int res = 0;for (int i = 0; i < n; ++i) {res += min(leftMax[i], rightMax[i]) - height[i];}return res;}
};

复杂度分析

时间复杂度: O ( n ) O(n) O(n)

空间复杂度: O ( n ) O(n) O(n)

方法二:单调栈

思路

除了计算并存储每个位置两边的最大高度以外,也可以用单调栈计算能接的雨水总量。

维护一个单调栈,单调栈存储的是下标,满足从栈底到栈顶的下标对应的数组 height 中的元素递减。

从左到右遍历数组,遍历到下标 i 时,如果栈内至少有两个元素,记栈顶元素为 toptop 的下面一个元素是 left,则一定有 height[left]≥height[top]。如果 height[i]>height[top],则得到一个可以接雨水的区域,该区域的宽度是 i−left−1,高度是 min⁡(height[left],height[i])−height[top],根据宽度和高度即可计算得到该区域能接的雨水量。

为了得到 left,需要将 top 出栈。在对 top 计算能接的雨水量之后,left 变成新的 top,重复上述操作,直到栈变为空,或者栈顶下标对应的 height 中的元素大于或等于 height[i]

在对下标 i 处计算能接的雨水量之后,将 i 入栈,继续遍历后面的下标,计算能接的雨水量。遍历结束之后即可得到能接的雨水总量。

代码

class Solution {
public:int trap(vector<int>& height) {int n = height.size(), res = 0;stack<int> stk;for (int i = 0; i < n; ++i) {while (!stk.empty() && height[i] > height[stk.top()]) {int top = stk.top(); stk.pop();if (stk.empty()) {break;}int left = stk.top();int curWidth = i - left - 1;int curHeigh = min(height[left], height[i]) - height[top];res += curWidth * curHeigh;}stk.push(i);}return res;}
};

复杂度分析

时间复杂度: O ( n ) O(n) O(n) n n n 是数组 height 的长度。从 0 到 n-1 的每个下标最多只会入栈和出栈一次。

空间复杂度: O ( n ) O(n) O(n)。空间复杂度主要取决于栈空间,栈的大小不会超过 n。

方法三:双指针

思路

下标 i 处能接的雨水量由 leftMax[i]rightMax[i] 中的最小值决定。数组 leftMax 是从左到右计算的,数组 rightMax 是从右到左计算的,我们可以使用双指针和两个变量来代替两个数组。

使用 height[left]height[right] 来更新 leftMaxrightMax 比较容易理解就不说了。重点说一下如何更新 res

  • height[left] < heigh[right] 时,必有 leftMax < rightMax,下标 left 处一定可以盛水;
  • height[left] >= heigh[right] 时,必有 leftMax >= rightMax,下标 right 处一定可以盛水。

代码

class Solution {
public:int trap(vector<int>& height) {int res = 0;int left = 0, right = height.size() - 1;int leftMax = 0, rightMax = 0;while (left < right) {leftMax = max(leftMax, height[left]);rightMax = max(rightMax, height[right]);if (height[left] < height[right]) {res += leftMax - height[left++]; }else {res += rightMax - height[right--];}}return res;}
};

复杂度分析

时间复杂度: O ( n ) O(n) O(n) n n n 为数组 height 的长度。

空间复杂度: O ( 1 ) O(1) O(1)


写在最后

如果您发现文章有任何错误或者对文章有任何疑问,欢迎私信博主或者在评论区指出 💬💬💬。

如果大家有更优的时间、空间复杂度的方法,欢迎评论区交流。

最后,感谢您的阅读,如果有所收获的话可以给我点一个 👍 哦。

这篇关于【面试经典 150 | 数组】接雨水的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!


原文地址:
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.chinasem.cn/article/951335

相关文章

Java中的数组与集合基本用法详解

《Java中的数组与集合基本用法详解》本文介绍了Java数组和集合框架的基础知识,数组部分涵盖了一维、二维及多维数组的声明、初始化、访问与遍历方法,以及Arrays类的常用操作,对Java数组与集合相... 目录一、Java数组基础1.1 数组结构概述1.2 一维数组1.2.1 声明与初始化1.2.2 访问

MySQL查询JSON数组字段包含特定字符串的方法

《MySQL查询JSON数组字段包含特定字符串的方法》在MySQL数据库中,当某个字段存储的是JSON数组,需要查询数组中包含特定字符串的记录时传统的LIKE语句无法直接使用,下面小编就为大家介绍两种... 目录问题背景解决方案对比1. 精确匹配方案(推荐)2. 模糊匹配方案参数化查询示例使用场景建议性能优

关于集合与数组转换实现方法

《关于集合与数组转换实现方法》:本文主要介绍关于集合与数组转换实现方法,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录1、Arrays.asList()1.1、方法作用1.2、内部实现1.3、修改元素的影响1.4、注意事项2、list.toArray()2.1、方

MySQL JSON 查询中的对象与数组技巧及查询示例

《MySQLJSON查询中的对象与数组技巧及查询示例》MySQL中JSON对象和JSON数组查询的详细介绍及带有WHERE条件的查询示例,本文给大家介绍的非常详细,mysqljson查询示例相关知... 目录jsON 对象查询1. JSON_CONTAINS2. JSON_EXTRACT3. JSON_TA

JAVA数组中五种常见排序方法整理汇总

《JAVA数组中五种常见排序方法整理汇总》本文给大家分享五种常用的Java数组排序方法整理,每种方法结合示例代码给大家介绍的非常详细,感兴趣的朋友跟随小编一起看看吧... 目录前言:法一:Arrays.sort()法二:冒泡排序法三:选择排序法四:反转排序法五:直接插入排序前言:几种常用的Java数组排序

Git可视化管理工具(SourceTree)使用操作大全经典

《Git可视化管理工具(SourceTree)使用操作大全经典》本文详细介绍了SourceTree作为Git可视化管理工具的常用操作,包括连接远程仓库、添加SSH密钥、克隆仓库、设置默认项目目录、代码... 目录前言:连接Gitee or github,获取代码:在SourceTree中添加SSH密钥:Cl

Java数组初始化的五种方式

《Java数组初始化的五种方式》数组是Java中最基础且常用的数据结构之一,其初始化方式多样且各具特点,本文详细讲解Java数组初始化的五种方式,分析其适用场景、优劣势对比及注意事项,帮助避免常见陷阱... 目录1. 静态初始化:简洁但固定代码示例核心特点适用场景注意事项2. 动态初始化:灵活但需手动管理代

数据库面试必备之MySQL中的乐观锁与悲观锁

《数据库面试必备之MySQL中的乐观锁与悲观锁》:本文主要介绍数据库面试必备之MySQL中乐观锁与悲观锁的相关资料,乐观锁适用于读多写少的场景,通过版本号检查避免冲突,而悲观锁适用于写多读少且对数... 目录一、引言二、乐观锁(一)原理(二)应用场景(三)示例代码三、悲观锁(一)原理(二)应用场景(三)示例

C++中初始化二维数组的几种常见方法

《C++中初始化二维数组的几种常见方法》本文详细介绍了在C++中初始化二维数组的不同方式,包括静态初始化、循环、全部为零、部分初始化、std::array和std::vector,以及std::vec... 目录1. 静态初始化2. 使用循环初始化3. 全部初始化为零4. 部分初始化5. 使用 std::a

shell编程之函数与数组的使用详解

《shell编程之函数与数组的使用详解》:本文主要介绍shell编程之函数与数组的使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教... 目录shell函数函数的用法俩个数求和系统资源监控并报警函数函数变量的作用范围函数的参数递归函数shell数组获取数组的长度读取某下的