首页 > 移动平台 > 详细

leetcode 42 Trapping Rain Water(Hard)

时间:2020-07-02 15:11:51      阅读:57      评论:0      收藏:0      [点我收藏+]

题目来源:leetcode 42 Trapping Rain Water

题目描述:

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
技术分享图片

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。?感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

解题思路:

可以拿总的(总长最高柱子的高度)-白色的(两个循环求得)-黑色的(每个柱子的高度1),就是雨水的量,具体见代码:

class Solution {
public:
    int trap(vector<int>& height) {
        int white=0,max=0,black=0,total=0;
        int size=height.size();
        int i;
        for(i=0;i<size;i++)
        {
            if(height[i]>max){
                white+=(height[i]-max)*i;
                max=height[i];
            }
        }
        max=0;
        for(i=size-1;i>=0;i--)
        {
            if(height[i]>max){
                white+=(height[i]-max)*(size-i-1);
                max=height[i];
            }
        }
        for(i=0;i<size;i++)
            black+=height[i];
        total=size*max;
        return total-black-white;
    }
};

leetcode 42 Trapping Rain Water(Hard)

原文:https://www.cnblogs.com/yjcoding/p/13224364.html

(0)
(0)
   
举报
评论 一句话评论(0
关于我们 - 联系我们 - 留言反馈 - 联系我们:wmxa8@hotmail.com
© 2014 bubuko.com 版权所有
打开技术之扣,分享程序人生!