给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水)。 感谢 Marcos 贡献此图。
示例:
输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
思路:
利用双指针遍历一次,通过分析可以得到
每一个位置存的水的面积 等于 其(左边最大高度和右边最大高度中那个较小的值)-当前高度
public int trap(int[] height) {
if (height == null || height.length <= 2)
return 0;
int left = 0;
int right = height.length - 1;
int leftMax = height[0];
int rightMax = height[height.length - 1];
int res = 0;
while (left <= right) { //要有等于 保证每一个位置都能遍历到
if(leftMax<rightMax){
//更新左边最大值
leftMax=Math.max(leftMax,height[left]);
res+=leftMax-height[left];
left++;
}else {
//更新右边最大值
rightMax=Math.max(rightMax,height[right]);
res+=rightMax-height[right];
right--;
}
}
return res;
}