LeetCode hot100 接雨水

最新推荐文章于 2024-11-01 15:19:21 发布

m0_67582670

最新推荐文章于 2024-11-01 15:19:21 发布

阅读量353

点赞数 7

分类专栏： leetcode 文章标签： leetcode 算法职场和发展

本文链接：https://blog.csdn.net/m0_67582670/article/details/134959536

版权

leetcode 专栏收录该内容

52 篇文章 0 订阅

订阅专栏

文章描述了一种算法，如何通过双指针法解决给定非负整数表示的柱子高度图问题，计算在降雨后能接收到的雨水总量。使用C++代码展示了两种实现方式：一种是逐次比较左右边界的最大值，另一种是同时更新左右指针指向的最大值。

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

题目

给定 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

提示：

n == height.length
1 <= n <= 2 * 104
0 <= height[i] <= 105

第一遍

第一次写hard，根本写不出来，网上看了思路，用双指针写，左边和右边分别向中间遍历，寻找最大值，装的水取决于左右两边已发现最大值的较小的一边。

class Solution {
public:
    int trap(vector<int>& height) {
        auto left_iter = height.begin();
        auto right_iter = height.end() - 1;
        int result = 0;
        int left_max = 0;
        int right_max = 0;
        while(left_iter <= right_iter){
            left_max = std::max(left_max, *left_iter);
            right_max = std::max(right_max, *right_iter);
            if(left_max < right_max){
                result += (left_max - *(left_iter));
                ++left_iter; 
            }
            else{
                result += (right_max - *(right_iter));
                --right_iter;
            }
        }
        return result;
    }
};

第二遍

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