题干
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。
思路:
计算一共可以接收多少单位的雨水,可以将问题拆分成每个横坐标柱子可以接雨水的总和,这样问题就变为每个柱子能接多少雨水。那么一个柱子能接多少雨水和那些因数有关呢?这时候我想到了水桶效应,所以可以想到影响的因素有该柱子左右两边最高柱子的高度,该柱子本身的高度。这样就可以把问题想成一个立方体的水槽前后的高度都无限高,两边的高度为左右两边最高柱子的高度,而水槽底部的高度为该柱子本身的高度,计算公式为
trap = min(leftHeight, rightHeight) - height
leftHeight: 该柱子左边的最高柱子高度
rightHeight:该柱子右边的最高柱子高度
height:该柱子本身的高度
到这里问题就变为怎么找到每个柱子左右两边最高柱子的高度,当然最容易想到的方法是左右两边遍历从而找到两边最高柱子的高度,不过这样的时间复杂度为O(n²)因为每个柱子都要进行全部遍历。这时想到了动态规划,定义两个数组leftHeight[length]和rightHeight[lenght],一个从左往右遍历,每个下标位置存储当前位置左边最高柱子高度,一个从右往左遍历,每个下标位置存储当前位置右边最高柱子高度。(以leftHeight[length]为例,当前位置的左边最高柱子高度等于相邻左边柱子高度与相邻柱子的左边最高柱子的高度两个值的最大值)
leftHeight[i] = max(leftHeight[i-1],height[i-1])
代码实现
public static int trap(int[] height) {
// 每个位置左边最大柱子高度数组
int[] leftHeight = new int[height.length];
// 0号位左边没有柱子则为0
leftHeight[0] = 0;
// 每个位置右边最大柱子高度数组
int[] rightHeight = new int[height.length];
// 最右边的位置的右边没有柱子,则设置为0
rightHeight[height.length - 1] = 0;
// 遍历保存每个位置左边最大柱子的高度
for (int item = 1; item < height.length; item ++) {
leftHeight[item] = Math.max(height[item - 1], leftHeight[item - 1]);
}
// 遍历保存每个位置右边最大柱子的高度
for (int item = height.length -2 ; item >= 0; item--) {
rightHeight[item] = Math.max(height[item + 1], rightHeight[item + 1]);
}
int sum = 0;
// 计算每个柱子能存储雨水的单位数量,并进行累计
for (int item = 0; item < height.length; item++) {
int diff = Math.min(leftHeight[item], rightHeight[item]) - height[item];
sum = sum + Math.max(diff, 0);
}
return sum;
}