接雨水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
示例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 个单位的雨水(蓝色部分表示雨水)。
示例 2:
输入:height = [4,2,0,3,2,5]
输出:9
class Solution {
public int trap(int[] height) {
int sum = 0;
int left = 0;
int right = height.length-1;
int leftMax = 0;
int rightMax = 0;
while(left<right){
leftMax = height[left]>leftMax?height[left]:leftMax;
rightMax = height[right]>rightMax?height[right]:rightMax;
if(leftMax<rightMax){
sum+=leftMax-height[left];
left++;
}else{
sum+=rightMax-height[right];
right--;
}
}
return sum;
}
}
解题思路
可以想象一下存在以下这么一个水渠,长度为n个单位,其中插入x(x<=n)块木板来拦截水渠存水,如果要知道其中存了多少水就要知道所有相隔两块木板之间存了多少水。
现在我们从两边往中间检查,发现左右各一块木板长度为1、1,如果我们现在想知道目前至少可以存多少水只要继续往中间检查一个长度,以当前木板长度减去下一个木板长度,则是下一个长度位置的存水量。
我们定义一个规则,左右最长木板长度相等时我们检查右边(其实左右都是可以的)。
我们在右边向内检查一个长度,右边发现长度为2的木板,可存水数可视为1-2<0为负数,我们不计入总数。且现在右边最长木板长度为2。
我们在左边继续检查一个长度,出现了没有插入木板的位置我们可以视为木板的长度为0,则目前至少可以存入水为1=1-0单位。
由于可存水数是取决于最短的木板,所有我们接下来应该从左边开始检查。
继续检查发现长度为2木板,可存水量为1-2<0,不计入总水量(sum = 1),左边最长木板长度为2。
右边继续检查长度为1木板,可存水量为1 = 2-1,计入总水量2=1+1。
右边继续检查,长度为2木板,可存水量为0= 2-2。不计入总水量(sum = 2)
右边继续检查,长度为3木板,可存水量为2-3<0,不计入总水量(sum = 2),最长木板长度为3。
左边继续检查,发现长度为1木板,可存水量为1 = 2-1,计入总水量3 = 2+1。
左边继续检查,发现长度为0木板,存水量为2= 2-0,计入总水量5 = 3+2。
左边继续检查,发现长度为1木板,存水量为1= 2-1,计入总水量6 = 5+1。
当左右两边相遇时检查结束。