能实现计算水滴数,但是不是遍历一遍数组和空间复杂度为O(1),我不清楚,懒得优化算法,代码仅仅作为参考。
代码如下 :
package HBase;
public class test {
public static void main(String[] args) {
int[] A = {1,0,2,1,0,1,3,2,1,2,1};
int waters = waterCount(A);
System.out.println(waters);
}
public static int waterCount(int[] A) {
int count = 0;
//for循环实现递归
for (int i = 0; i < A.length; i++) {
int beginHeight = A[i]; //设置桶的左边
for (int j = i + 1; j < A.length; j++) {
int endHeight = A[j]; //设置桶的右边
if (beginHeight <= endHeight) { //找到桶的右边比桶的左边高的位置。
int initHeight = beginHeight;
//计算区间内可容纳水滴数
for (int k = i + 1; k < j; k++) {
int waters = initHeight - A[k];
count = count + waters;
}
//移动桶的左边到桶的右边前一个位置
i = j - 1;
break;
} else if (j == A.length - 1) {
break;
}
}
}
//输出水滴数
return count;
}
}