接雨水
题意:
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [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
输出:
9
解题思路:
每一个位置能乘多少水取决于,当前位置左边的最大高度leftMaxHeight和右边的最大高度rightMaxHeight的较小值减去当前高度。如果暴力的话,复杂度是o(n2),这里我们可以灵活一点提前记录好每个位置左边的最大值和右边的最大值,这样时间复杂度降到o(n),空间复杂度为o(n)。
// 每一个位置能乘多少水取决于,当前位置左边的最大高度leftMaxHeight和右边的最大高度rightMaxHeight的较小值减去当前高度
public static int trap(int[] height) {
if (null == height || 0 == height.length) {
return 0;
}
int count = 0;
final int size = height.length;
// 记录一下当前位置左边的最大值
int[] dpLeft = new int[size];
// 记录一下当前位置右边的最大值
int[] dpRight = new int[size];
for(int i = 1 ; i < size ; i++ ){
dpLeft[i] = Math.max(dpLeft[i-1] , height[i-1]);
}
for(int i = size-2 ; i > 0 ; i--){
dpRight[i] = Math.max(dpRight[i+1] , height[i+1]);
}
for(int i = 1; i < size-1;i++){
count += Math.max(Math.min(dpLeft[i] , dpRight[i]) - height[i] , 0);
}
return count;
}