Trapping Rain Water

最新推荐文章于 2024-09-17 16:23:14 发布

wickedvalley

最新推荐文章于 2024-09-17 16:23:14 发布

阅读量203

点赞数

CC 4.0 BY-SA版权

分类专栏：笔试编程文章标签：编程

本文链接：https://blog.youkuaiyun.com/wickedvalley/article/details/74809163

笔试编程专栏收录该内容

77 篇文章

订阅专栏

本文介绍了一种计算雨水收集量的算法。通过寻找数组中的最高点，并分别从左右两侧计算可以存储的雨水总量。此方法适用于非负整数数组表示的高度图。

摘要生成于 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.

For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

解析思路：先找到最高峰，然后从两边向中间计算，因为两边向中间是递增的趋势

参考代码如下：

    int trap(int A[], int n) {
        if(n <= 2) return 0;
        int max = -1, maxInd = 0;
        int i = 0;
        for(; i < n; ++i){//先找出最高峰
            if(A[i] > max){
                max = A[i];
                maxInd = i;
            }
        }
        int area = 0, root = A[0];
        for(i = 0; i < maxInd; ++i){//从左侧到最高峰的计算
            if(root < A[i]) root = A[i];
            else area += (root - A[i]);
        }
        for(i = n-1, root = A[n-1]; i > maxInd; --i){//从右侧到最高峰的计算
            if(root < A[i]) root = A[i];
            else area += (root - A[i]);
        }
        return area;
    }