建立一个数组num,表示当前位置的水平高度(包括水和bar之和)。第一个位置num[0]和最后一个位置num[n-1]因为是开区间,肯定不能trap water。
从第二个位置num[1]开始往后扫描,如果num[i]取num[i-1]和A[i]的较大者。表示该位置水平高度的上限(尚未考虑右侧邻居)。
然后从倒数第二个位置开始往前扫描,保证num[i] <= num[i+1],并且num[i]不小于A[i]。
每个位置trap的水量即为num[i]-A[i]。
时间O(n),空间O(n)。
class Solution {
public:
int trap(int A[], int n) {
int *num = new int[n];
memset(num, 0, sizeof(int)*n);
int total = 0;
num[0] = A[0];
num[n-1] = A[n-1];
for(int i=1; i<n-1; i++)
num[i] = max(A[i], num[i-1]);
for(int i=n-2; i>=1; i--)
{
num[i] = max(A[i], min(num[i], num[i+1]));
total += (num[i]-A[i]);
}
delete []num;
return total;
}
};