思路:首先找到最高的那个bar,然后分别从左右两端往中间最高的bar遍历。3个for循环,复杂度为O(n)。
左边往中间遍历时,遍历到第i个位置时,这个bar能存储的水量是leftmax-A[i],leftmax是之前遍历的bar(包括第i个位置)中最高的那个值。
同理,右边往中间遍历时也是如此。
注意当数组长度小于3时,return 0.
public int trap(int[] A) {
int n=A.length;
if(n<3)return 0;
int maxIndex = 0,maxValue=0;
int sum =0;
for(int i=0;i<n;i++){
if(maxValue<A[i]){
maxIndex = i;
maxValue = A[i];
}
}
int left=1,right=n-2;
int leftmax=A[0],rightmax=A[n-1];
for(int i=1;i<maxIndex;i++){
if(A[i]>leftmax){
leftmax = A[i];
}
sum += leftmax-A[i];
}
for(int i=n-2;i>maxIndex;i--){
if(A[i]>rightmax){
rightmax = A[i];
}
sum += rightmax-A[i];
}
return sum;
}