【字节专题】leetcode 42. 接雨水

最新推荐文章于 2024-04-09 09:30:40 发布

Elon0

最新推荐文章于 2024-04-09 09:30:40 发布

阅读量206

点赞数

分类专栏：字节LeetCode专场

本文链接：https://blog.csdn.net/biggoodloong/article/details/102098441

版权

这篇博客探讨了LeetCode第42题——接雨水的问题，通过动态规划和栈两种方法进行解决。在动态规划解决方案中，使用max_left和max_right数组分别记录每列左边和右边的最高墙。在栈解决方案中，当当前高度小于等于栈顶高度时入栈，否则出栈并计算水的体积，直至栈空或当前高度小于等于栈顶高度。

摘要由CSDN通过智能技术生成

42. 接雨水
在这里插入图片描述

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

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 感谢 Marcos 贡献此图。

示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

逻辑：

当前高度小于等于栈顶高度，入栈，指针后移。
当前高度大于栈顶高度，出栈，计算出当前墙和栈顶的墙之间水的多少，然后计算当前的高度和新栈的高度的关系，重复第 2 步。直到当前墙的高度不大于栈顶高度或者栈空，然后把当前墙入栈，指针后移。

动态规划

首先用两个数组，max_left [i] 代表第 i 列左边最高的墙的高度，max_right[i] 代表第 i 列右边最高的墙的高度。（一定要注意下，第 i 列左（右）边最高的墙，是不包括自身的，和 leetcode 上边的讲的有些不同）

public int trap(int[] height) {
   
    int sum = 0;
    int[] max_left = new int[height.length];
    int[] max_right = new int[height.length];

    for (int i