Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.
左右两个指针往中间遍历,记录对应左右两边的最大值更新,移动最大值小的那一边,这样可以只遍历数组一遍。
public class Solution {
//Time: O(n)
public int trap(int[] A) {
if (A == null || A.length <= 1) {
return 0;
}
int left = 0;
int leftmax = A[left];
int right = A.length - 1;
int rightmax = A[right];
int sum = 0;
while (left < right) {
if (rightmax > leftmax) {
left++;
sum += leftmax - A[left] > 0 ? leftmax - A[left] : 0;
leftmax = Math.max(leftmax, A[left]);
} else {
right--;
sum += rightmax - A[right] > 0 ? rightmax - A[right] : 0;
rightmax = Math.max(rightmax, A[right]);
}
}
return sum;
}
}
No comments:
Post a Comment