给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例 1:
输入:height = [0,1,0,2,1,0,1,,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
记录一下官方动态规划的解法:
- 定义 leftMaxArray rightMaxArray 两数组 (leftMaxArray 用来表示从 [0, i] 处柱子的最高高度,rightMaxArray 表示从 [i, height.length-1] 处柱子的最高高度(即两数组分别记录当前索引处向两边延申可获取到的最高边界柱值))。
- 分别从左向右 / 从右向左遍历数组初始化两记录数组。
- 遍历数组,当前下标处可储存的雨水量为 两数组高度的最小值(即有效柱高)-当前下标处柱高
public int trap(int[] height) {
int result = 0, i;
int length = height.length;
if (length < 2) return result;
int[] leftMaxArray = new int[length], rightMaxArray = new int[length];
leftMaxArray[0] = height[0];
for (i = 1; i < length; ++ i) {
leftMaxArray[i] = Math.max(height[i], leftMaxArray[i-1]);
}
rightMaxArray[length-1] = height[length-1];
for (i = length-2; i >= 0; -- i) {
rightMaxArray[i] = Math.max(height[i], rightMaxArray[i+1]);
}
for (i = 0; i < length; ++ i) {
result += Math.min(leftMaxArray[i], rightMaxArray[i])-height[i];
}
return result;
}