给定 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
批注:虽然leetcode上标注为hard,但只要把过程想清楚,实现起来其实很简单。
下面是一个直接的方法:
分析每一列,当左边且右边的最大的列大于这一列时,这一列上才可能存水。
class Solution {
public:
int trap(vector<int>& height) {
int coun=0;
for(int i=1;i<height.size();i++)
{
int max_left=0;
int max_right=0;
for(int j=0;j<i;j++)
max_left=height[j]>max_left?height[j]:max_left;
for(int j=i+1;j<height.size();j++)
max_right=height[j]>max_right?height[j]:max_right;
if(max_left<=height[i]||max_right<=height[i])
coun+=0;
else
coun+=(min(max_right,max_left)-height[i]);
}return coun;
}
};