题目:给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
图示:
描述:蓝色表示雨水,黑色表示柱子,上面的图片说明这组数据可以接6格雨水
(来源:https://leetcode-cn.com/problems/trapping-rain-water/)
思路:按照问题描述,要知道这个区域能容纳多少雨水,其值等于这一格两边高度较小的减去当前格的高度值。只需要从第一格开始(下标为1)往两边寻找,得出每格能装多少雨水即可。
代码:
int trap(int* height, int heightSize)
{
int v=0,min;
int i,j,p=0,q=0;
for(i=1;i<heightSize-1;i++)
{
p=0;
q=0;
for(j=i;j>=0;j--)
p=p>height[j]?p:height[j];
for(j=i;j<heightSize;j++)
q=q>height[j]?q:height[j];
min=p<q?p:q;
v+=min-height[i];
}
return v;
}