给定n代表高度图的非负整数,其中每个小节的宽度为1,计算下雨后能捕捉多少水。
例如,给定[0,1,0,2,1,0,1,3,2,1,2,1],返回6。
解题思想:
扫描一遍,找到最高的柱子,这个柱子将数组分为两半;
处理左边一半;
处理右边一半
在其一半之中定义一个水位,每次利用水位差来计算该位置所存的水滴数量,最后water+=1,计算总和。
代码:
package ACM;
public class Solution {
public static int trap(int[] A) {
int n = A.length;
int height = 0;
for (int i = 0; i < n; i++)
if (A[i] > A[height]) height = i;
int water = 0;
for (int i = 0, index = 0; i < height; i++) {
if (A[i] > index) index = A[i];
else water += index - A[i];
}
for (int i = n - 1, index = 0; i > height; i--) {
if (A[i] > index) index = A[i];
else water += index - A[i];
}
return water;
}
public static void main(String[] args) {
int[] A={0,1,0,2,1,0,1,3,2,1,2,1};
System.out.println(trap(A));
}
};