算法：接雨水

最新推荐文章于 2025-07-13 15:31:26 发布

晋阳

最新推荐文章于 2025-07-13 15:31:26 发布

阅读量2.4k

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签：算法 LeetCode JAVA

本文链接：https://blog.youkuaiyun.com/weixin_40564421/article/details/79049100

算法专栏收录该内容

8 篇文章

订阅专栏

本文介绍了一个经典的算法问题——接雨水问题，并提供了一种详细的解决方案。该问题要求计算一系列非负整数代表的高度图中能够存储多少雨水。文章通过示例说明了如何逐层计算并累加每一层可以接收的雨水量。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

问题

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

样例

接雨水

如上图所示，海拔分别为 [0,1,0,2,1,0,1,3,2,1,2,1], 返回 6.

思路

思路是先看最底下一层能接多少雨水，然后再依次推断到第二层，第三层......最高层。

实现

public static int trapRainWater(int[] heights) {
        // write your code here
        // 先找到最大值
        int max = 0;
        for (int i=0;i<heights.length;i++) {
            max = Math.max(max, heights[i]);
        }
        int rains = 0;
        // 思路是先看最底下一层能接多少雨水，然后再依次推断到最高层
        for (int floor = 0; floor < max; floor++) {
            // 当层数为floor层时
            // 从第1位到第一个非0位之间的0都置为1（便于统计这一层的雨水）
            for (int i=0;i<=heights.length-1;i++) {//统一使用0-start
                if (heights[i] == 0) {
                    heights[i] = 1;
                } else {
                    break;
                }
            }
            // 从最后1位到倒数第一个非0位之间的0都置为1（便于统计这一层的雨水）
            for (int i=heights.length-1;i>=0;i--) {//统一使用0-start
                if (heights[i] == 0) {
                    heights[i] = 1;
                } else {
                    break;
                }
            }
            for (int i=0;i<heights.length;i++) {
                if (heights[i] == 0) {
                    //经过前面的准备工作，我们可以很轻松的分辨出来哪些是雨水
                    rains++;
                } else {
                    //这里要做个操作，就是把数组里的所有元素都减1，为了上面一层的计算做准备
                    heights[i]--;
                }
            }

        }
        return rains;
    }