题目
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
输入:height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出:6
解释:上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图,在这种情况下,可以接 6 个单位的雨水(蓝色部分表示雨水),详细见下图。
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/trapping-rain-water
分析
这道题可以用动态规划来解题,但最简单的还是双指针,也最容易理解,关键是执行速率最快,第一次击败100%的用户,标题特别标注庆祝一下,哈哈........
接雨水的多少取决于桶的高度,桶的高度由桶两边的最小值决定,所以双指针在往中间靠拢时,更矮的优先移动。
每往中间靠拢一次就需要判断要不要更新桶高度,如果比上一次桶更小则不需要。
代码
class Solution {
public int trap(int[] height) {
int left =0,right = height.length-1;
int result = 0;
int h = 0;
while(left < right) {
if(height[left] < height[right]){
if(height[left] < h){
result += h - height[left];
}
h = Math.max(Math.min(height[left], height[right]), h);
left ++;
}else {
if(height[right] < h){
result += h - height[right];
}
h = Math.max(Math.min(height[left], height[right]), h);
right --;
}
}
return result;
}
}
结果
关注个人微信公众号:肌肉码农,回复“好友”讨论问题。