42 接雨水

最新推荐文章于 2024-07-16 23:15:32 发布

老宋啊

最新推荐文章于 2024-07-16 23:15:32 发布

阅读量56

点赞数

文章标签： leetcode 算法数据结构

本文链接：https://blog.csdn.net/qq_34473348/article/details/125244901

版权

42 接雨水

题目：给定 n 个非负整数表示每个宽度为 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 个单位的雨水（蓝色部分表示雨水）。

思考：

双指针
桶高

public class Solution {
    /**
     * 接雨水
     * 输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
     * 输出：6
     */
    public int trap(int[] arr) {
        // 双指针
        int left = 0, right = arr.length - 1;
        // 桶高
        int height = 0;
        // 如果返回值是long，那么雨水量类型就是long
        // 本题是int类型
        int water = 0;

        while (left < right) {
            // 桶高：左右指针、历史桶高的最大值
            height = Math.max(height, Math.min(arr[left], arr[right]));

            // 雨水量 += 最低高度 - 左右最矮
            water += arr[left] <= arr[right] ? (height - arr[left++]) : (height - arr[right--]);
        }
        return water;
    }

    public static void main(String[] args) {
        Solution solution = new Solution();
        int[] nums = {0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1};
        System.out.println(solution.trap(nums));
    }
}