42. 接雨水
给定 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
class Solution {
public int trap(int[] height) {
int n = height.length, sum = 0;
if(n == 0) return 0;
int[] left = new int[n];
int[] right = new int[n];
left[0] = height[0];
right[n-1] = height[n-1];
/*思路:动态规划求解
* 计算出每一列中左边最高, 右边最高的高度
* 最后每一列中包含水滴的个数即为两边最低的高度减去当前高度
*/
for(int i = 1; i < n; i ++) {
left[i] = Math.max(height[i], left[i-1]);
}
for(int i = n-2; i >= 0; i --) {
right[i] = Math.max(height[i], right[i+1]);
}
//计算每一列中水滴的数量
for(int i = 0; i < n; i ++) {
sum += Math.min(left[i], right[i])-height[i];
// System.out.println(Math.min(left[i], right[i])-height[i]);
}
return sum;
}
}