题目:
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.
For example,
Given [0,1,0,2,1,0,1,3,2,1,2,1]
, return 6
.
思路:
这道题其实很多地方都有过(印象中是在刷LeetCode之前做LintCode的时候出现过的),当时的方法是用一前一后两个标记位,标记当前水池两端的高度。两边高度较小的一方即表示当前水池能装水的最高处。
比如当前左边高度是1,右边高度是3,那么两端最小部分是1,那么从左边开始一直向右边遍历,直到遍历到某个一处高度大于或等于1停止,那么遍历过程中经过的全部位置,均可装深度为1的水。之后重新判断新的左右两端哪边更小,从而决定是从左往右遍历还是从右往左遍历。
代码:
class Solution {
public int trap(int[] height) {
int result = 0;
int left = 0,right = height.length - 1;
while(left < right){
int min = Math.min(height[left],height[right]);
if(height[left] == min){
while(height[++left] < min && left < right)
result += (min - height[left]);
}else{
while(height[--right] < min && left < right)
result += (min - height[right]);
}
}
return result;
}
}