力扣第42题接雨水

news2024/9/20 0:50:41

前言

记录一下刷题历程力扣第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

分析

我们首先可以想到的就是统计一列左边和右边所有柱子的最高高度，根据下图我们可以知道左右最高高度较低的值与当前柱子的高度差就是这一列能存储的水的高度。这是第一种方法但是需要O(n)的空间复杂度，有没有优化空间呢，我们来看方法2.
在这里插入图片描述
第二种方法，我们其实并不需要同时计算左右两边的最高高度，假如说我们对于当前列已经知道了左侧的最高高度，只要右侧出现比左侧高度还要高，那么左侧的最高高度就一定是水体的顶部，然后我们可以计算该列的水体高度，我们从数组两端向中间遍历，分别计算左指针和左侧最高高度，右指针和右侧最高高度，找到对应结果较低的那一列直接更新结果即可

代码如下：

第一种方法：

class Solution {
public:
    // 接收一个整数数组 height，表示每个柱子的高度，返回可以接住的雨水总量
    int trap(vector<int>& height) {
        // n 表示柱子的总数
        int n = height.size();
        // res 用于存储最终结果，也就是可以接住的雨水总量
        int res = 0;

        // 创建两个数组，分别存储每个位置的左边最高柱子和右边最高柱子
        vector<int> leftMax(n);  // leftMax[i] 表示位置 i 左边最高的柱子
        vector<int> rightMax(n); // rightMax[i] 表示位置 i 右边最高的柱子

        // 初始化 leftMax 数组的第一个元素，因为第 0 个位置没有左边的柱子，所以 leftMax[0] 就是 height[0]
        leftMax[0] = height[0];

        // 计算每个位置的左边最高柱子，从左到右遍历
        for(int i = 1; i < n - 1; i++) {
            // leftMax[i] 是当前位置 i 和位置 i-1 的最高柱子之间的较大值
            leftMax[i] = max(leftMax[i - 1], height[i]);
        }

        // 初始化 rightMax 数组的最后一个元素，因为最后一个位置没有右边的柱子，所以 rightMax[n-1] 就是 height[n-1]
        rightMax[n - 1] = height[n - 1];

        // 计算每个位置的右边最高柱子，从右到左遍历
        for(int i = n - 2; i > 0; i--) {
            // rightMax[i] 是当前位置 i 和位置 i+1 的最高柱子之间的较大值
            rightMax[i] = max(rightMax[i + 1], height[i]);
        }

        // 遍历每个柱子，计算该柱子上方能够接住的雨水
        for(int i = 1; i < n - 1; i++) {
            // 当前位置的雨水容量等于左边和右边最高柱子之间的较小值减去当前柱子的高度
            int capacity = min(leftMax[i], rightMax[i]) - height[i];
            // 累加结果，capacity 是该位置能接住的水
            res += capacity;
        }

        // 返回最终的接水量
        return res;
    }
};

第二种方法：

class Solution {
public:
    int trap(vector<int>& height) {
        // 获取柱子的总数
        int n = height.size();
        // 如果柱子数小于3，无法接住雨水，直接返回0
        if (n < 3) return 0;
        
        // 结果变量，用于存储接住的雨水总量
        int res = 0;
        // 左指针初始化为数组的起始位置
        int left = 0;
        // 右指针初始化为数组的结束位置
        int right = n - 1;
        // 左边最大值初始化为0
        int leftMax = 0;
        // 右边最大值初始化为0
        int rightMax = 0;
        
        // 双指针遍历，直到两个指针重合
        while (left < right) {
            // 更新左边最大值
            leftMax = max(leftMax, height[left]);
            // 更新右边最大值
            rightMax = max(rightMax, height[right]);
            
            // 如果左边最大值小于右边最大值，处理左边的情况
            if (leftMax < rightMax) {
                // 当前位置的水量 = 左边最大值 - 当前柱子的高度
                res += leftMax - height[left];
                // 左指针向右移动
                left++;
            } 
            // 如果右边最大值小于等于左边最大值，处理右边的情况
            else {
                // 当前位置的水量 = 右边最大值 - 当前柱子的高度
                res += rightMax - height[right];
                // 右指针向左移动
                right--;
            }
        }
        // 返回接住的雨水总量
        return res;
    }
};

代码解释

第一种方法主要思路：

1.对于每个柱子而言，能接住的水量取决于左边最高的柱子和右边最高的柱子，以及当前位置柱子的高度。柱子能接住的水量是左、右柱子中较矮的那个减去当前柱子的高度。

为了快速得到每个柱子左边和右边的最大值，我们通过预处理分别计算 leftMax 和 rightMax 数组。然后对于每个柱子，计算它上方可以接住的水量并累加到结果中。

2.首先，我们用两个数组 leftMax 和 rightMax 分别记录每个柱子左边和右边的最高柱子。

leftMax[i] 表示位置 i 左边的最高柱子，包括 i 本身。
rightMax[i] 表示位置 i 右边的最高柱子，包括 i 本身。

3.遍历 height 数组计算每个位置的左边最高值和右边最高值。

对于每个位置的柱子，我们计算其能接住的雨水量：min(leftMax[i], rightMax[i]) - height[i]。
累加每个位置的水量得到最终结果。

复杂度分析：

时间复杂度：该算法需要遍历三次数组，时间复杂度为 O(n)。
空间复杂度：需要额外的两个数组存储左边和右边的最大值，空间复杂度为 O(n)。
这就是该算法的基本思路和实现方式。

第二种方法主要思路：

1.初始化：

获取 height 数组的大小 n，存储在变量 n 中。
如果柱子的数量小于3（例如0、1或2个柱子），无法形成凹槽来接雨水，直接返回0。
初始化结果变量 res 为0，用于存储计算得到的雨水总量。
初始化两个指针 left 和 right 分别指向数组的起始和结束位置。
初始化 leftMax 和 rightMax 为0，用于记录当前左边和右边的最大高度。