接 雨 水
给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。
暴力法(对于数组中的每个元素,我们找出下雨后水能达到的最高位置,等于两边最大高度的较小值减去当前高度的值。):
class Solution {
public int trap(int[] height) {
int res=0;
int len =height.length;
//数组第一个数不用计算,左边无数,无法存水。
for(int i= 1; i< len-1 ;i++){
int leftmax =0;
int rightmax =0;
//搜索左边最大值
for(int j = i ; j>=0 ; j-- ){
leftmax=Math.max(leftmax,height[j]);
}
//搜索右边最大值
for(int j =i ;j<len ; j++){
rightmax=Math.max(rightmax,height[j]);
}
res=res+Math.min(leftmax,rightmax)-height[i];
}
return res;
}
}
双指针法
(从上述方法的示意图中我们注意到,只要 rightmax>leftmax(元素 0 到元素 6),积水高度将由 leftmax 决定,类似地 leftmax>rightmax(元素 8 到元素 11)。所以我们可以认为如果一端有更高的条形块(例如右端),积水的高度依赖于当前方向的高度(从左到右)。当我们发现另一侧(右侧)的条形块高度不是最高的,我们则开始从相反的方向遍历(从右到左):
class Solution {
public int trap(int[] height) {
int res=0;
//定义双指针
int left =0 ,right =height.length-1;
//定义左右最大值
int leftmax=0, rightmax=0;
while(left < right){
//如果右边最大值比左边最大值大(右边高度高),那么积水的高度由左边最大值决定。
if(height[left] < height[right]){
//如果遇到了比左边最大值大的数,更新左边最大值
if(height[left] > leftmax){
leftmax=height[left];
//当前位置比左边最大值小,那么说明可以存水。
}else{
res=res+(leftmax - height[left]);
}
//指针右移
left++;
//如果左边最大值比右边最大值大(左边高度高),那么积水的高度由右边最大值决定。
}else{
//如果遇到了比右边最大值大的数,更新右边最大值
if(height[right]>rightmax){
rightmax=height[right];
//当前位置比右边最大值小,那么说明可以存水。
}else{
res=res+ (rightmax - height[right]);
}
//指针向左移
right--;
}
}
return res;
}
}