给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水
上面是由数组 [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
题解:
暴力法
思路 从第二个 到倒数第二个 每一个位置能接受的雨水数量是
左边最高 和右边最高 leftMax rightMax 两者较小的 那个 和自己的高度差
比如 第二个位置 角标为1 高度为0 leftMax为1 rightMax为3 较小的是1 和自己高度差是1
所以第二个位置能接的雨水是1
public int trap(int[] height) {
int count = 0;
for(int i = 1;i<height.length-1;i++){
int leftMax = 0;
int rightMax = 0;
for(int j=0;j<i;j++){
leftMax = Math.max(leftMax,height[j]);
}
for(int k=i;k<height.length;k++){
rightMax = Math.max(rightMax,height[k]);
}
int i1 = Math.min(rightMax, leftMax) - height[i];
count += i1>0?i1:0;
}
return count;
}
动态编程
思路:先遍历一遍 用一个数组每个位置左边最大的数记录下来
同时用另一个数组把每个位置右边最大的数记录下来
再遍历数组 得到每个位置的左右最大数 之后计算
int max = 0;
if(height==null ||height.length==0 ){
return max;
}
int[] leftmax = new int[height.length-1];
int[] rightmax = new int[height.length-1];
int left = 0;
int right = 0;
for (int i = 1; i < height.length-1; i++) {
left = Math.max(height[i-1], left);
leftmax[i] = left;
}
for (int i = height.length-2; i >0; i--) {
right = Math.max(height[i+1], right);
rightmax[i] = right;
}
for (int i = 1; i < height.length-1; i++) {
int i1 = Math.min(leftmax[i], rightmax[i]) - height[i];
if(i1>0){
max +=i1;
}
}
return max;
双指针
上面思想是用两个数组维护每个位置左右最大值
那么我们可以换一种思路 因为我们只需要拿到左右最大值中比较小的那个就行了
举个例子:
第一步 右边的最大值比较小 所以 先走右指针 当前位置 能存水2-1
第二步 右边的最大值比较小 所以 继续走右指针 当前位置 能存水2-2
第N步 直到左边比较大 移动左边
总体思路是无论如何移动 总能得到两边最大值较小的那个
public int trap(int[] height) {
if(height ==null || height.length<3){
return 0;
}
int result= 0;
int leftMax= 0;
int rightMax=0;
int left = 0;
int right = height.length-1;
while(left<right){
leftMax = Math.max(leftMax,height[left]);
rightMax = Math.max(rightMax,height[right]);
if(leftMax>rightMax){
result += rightMax>height[right]?rightMax-height[right]:0;
right--;
}else{
result += leftMax>height[left]?leftMax-height[left]:0;
left ++;
}
}
return result;
}