题目来源: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;
}
};