给定 n 个非负整数表示每个柱的宽度为1的高程图,计算下雨后能够捕获的水量。
例如,给定 [0, 1, 0, 2, 1, 0, 1, 3, 2, 1, 2, 1],return 6 。
解题思路:
- 为了保证计算的结果有效,首先找出数组中最大值,
- 从最大值的两侧出发,分两部分计算
- 以左边为例:依次比较该元素与右边元素的大小,如果该元素大于右边的元素,则做差并记录结果,因为最大值在中间,所以这个结果肯定有效;如果该元素小于右边的元素,则进行替换,将该元素替换成右边的元素,依次进行。
代码如下:
public int trap(int[] A) {
int sum = 0;
// 找到数组中的最大值
int maxIndex = 0;
for(int i = 0; i < A.length; i++){
if(A[maxIndex] < A[i]){
maxIndex = i;
}
}
int left = A[0];
int right = A[A.length - 1];
// 依次比较并计算结果
for(int i = 1; i < maxIndex; i++){
// 如果 left > 该元素,记录差值,插值即为当前的积水量
if(left > A[i]){
sum += left - A[i];
}
// 如果小于该元素,更换 left 值
else{
left = A[i];
}
}
for(int i = A.length - 2; i > maxIndex; i--){
if(right > A[i]){
sum += right - A[i];
}
else
right = A[i];
}
return sum;
}