题目描述
给定 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:动态编程,从左往右遍历一次,记录当前位置左边最大值,再从右向左遍历一次记录当前位置右边最大值,则当前位置可以接的雨水为(min(左边最大值,右边最大值)- height),
即得到动态方程为:count += (Math.min(lMax[i], rMax[i]) - height[i]);
public int trap(int[] height) {
int[] lMax = new int[height.length];
int[] rMax = new int[height.length];
int max = 0;
for (int i = 0; i < height.length; i++) {
max = Math.max(max, height[i]);
lMax[i] = max;
}
max = 0;
for (int i = height.length-1; i >= 0; i--) {
max = Math.max(max, height[i]);
rMax[i] = max;
}
int count = 0;
for (int i = 0; i < height.length; i++) {
count += (Math.min(lMax[i], rMax[i]) - height[i]);
}
return count;
}