37_leetcode_Trapping Rain water

雨水存储算法

最新推荐文章于 2025-09-13 14:08:34 发布

ezzffzz

最新推荐文章于 2025-09-13 14:08:34 发布

阅读量314

点赞数

CC 4.0 BY-SA版权

分类专栏： leetcode 算法编程文章标签： leetcode 算法编程语言面试 string

本文链接：https://blog.youkuaiyun.com/ezzffzz/article/details/32299703

leetcode 同时被 3 个专栏收录

155 篇文章

订阅专栏

算法

155 篇文章

订阅专栏

编程

155 篇文章

订阅专栏

本文介绍了一种计算特定地形中雨水存储量的算法。该算法通过两次遍历数组来确定左右两侧的最大高度，进而计算出能够存储的雨水量。适用于非负整数数组表示的地形图。

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.

1:特殊情况，A==NULL, n <= 2；2:遍历数组获得值最大的Index；3:分别从左到右和从右到左遍历到最大的index处，在遍历的过程中，找到当前最大的值的index，并计算可以储存的水的数量，并随时更新当前最大值的索引

 int trap(int A[], int n) {
        
        if( A == NULL || n <= 2)
        {
            return 0;
        }
        
        int maxIndex = 0;
        for(int i = 0; i < n; i++)
        {
            if(A[i] > A[maxIndex])
            {
                maxIndex = i;
            }
        }
        
        int water = 0;
        
        int leftMaxIndex = 0;
        int tempIndex = leftMaxIndex + 1;
        while(tempIndex <= maxIndex - 1)
        {
            if(A[tempIndex] < A[leftMaxIndex])
            {
                water += A[leftMaxIndex] - A[tempIndex];
            }
            else
            {
                leftMaxIndex = tempIndex;
            }
            tempIndex++;
        }
        
        int rightMaxIndex = n - 1;
        tempIndex = rightMaxIndex - 1;
        while(tempIndex >= maxIndex + 1)
        {
            if(A[tempIndex] < A[rightMaxIndex])
            {
                water += A[rightMaxIndex] - A[tempIndex];
            }
            else
            {
                rightMaxIndex = tempIndex;
            }
            tempIndex--;
        }
        
        return water;
    }