Leetcode: 42. Trapping Rain Water
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.
Example:
Input: [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6
Solution:
class Solution {
public int trap(int[] height) {
if(height == null || height.length == 0)
return 0;
int leftIndex = 0, rightIndex = height.length - 1, leftMax = 0, rightMax = 0, result = 0;
while(leftIndex < rightIndex) {
leftMax = Math.max(leftMax, height[leftIndex]);
rightMax = Math.max(rightMax, height[rightIndex]);
if(leftMax < rightMax) {
result += leftMax - height[leftIndex];
leftIndex++;
}
else {
result += rightMax - height[rightIndex];
rightIndex--;
}
}
return result;
}
}
网友评论