LeetCode--接雨水(Trapping Rain Water 积水问题)

接雨水

题目描述 : 给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水

 

上面是由数组 [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

解题思路 : 在这里我们只需要解决最高峰问题即可 ; 首先可以定义两个常量, 一个 left 代表数组最左边的位置, 一个 right 代表数组最右边的位置, 然后再用一个 min记录下标left 和 right中数组的最小值 ; 假如min为下标 left 所表示数组大小, 然后再右移数组判断判断该元素是否小于min且left < right , 若小于则记录两者的差值, 若大于则更新min, 使min表示下标left 和 right中数组的最小值 ; 假如min为下标 right所表示数组的大小 , 同理 , 最后当 left == right的时候, 就是数组中的最高峰的下标;

要记住始终保持用较小值去判断 , 这样才不会造成遗漏;

扫描二维码关注公众号,回复: 2984510 查看本文章
int trap(int* height, int heightSize) {

    int l=0;
    int r=0;
    int min=0;
    int total=0;
    l=0;
    r=heightSize-1;
    while(l<r)
    {
        //min取其中的最小值
        min=height[l]<height[r]?height[l]:height[r];
        if(min==height[l])
        {
            l++;
            while(l<r&&min>height[l])
                total+=min-height[l++];
        }
        else
        {
            r--;
            while(l<r&&min>height[r])
                total+=min-height[r--];        
        }  
    }
    return total;
}

猜你喜欢

转载自blog.csdn.net/ds19980228/article/details/81783760