题目
思路
-
暴力
每次都只算指针p柱子上的水
向左找一个最大柱子,向右找一个最大的柱子,然后找出这俩柱子里最小的,算出和p柱子的差值(如果负数,说明p柱子很高,所以没有雨水)
代码(暴力)
class Solution {
public int trap(int[] height) {
int sum = 0;
for (int i = 1; i < height.length; i++){
int leftHeight = 0;
int rightHeight = 0;
//找左边最大
for (int left = i - 1;left >= 0; left --){
leftHeight = Math.max(leftHeight, height[left]);
}
//找右边最大
for (int right = i + 1; right < height.length; right++){
rightHeight = Math.max(rightHeight, height[right]);
}
int h = Math.min(leftHeight , rightHeight) - height[i];
if (h < 0){
continue;
}
sum += h;
}
return sum;
}
}
代码(优化)
核心思想:
双指针
只有低的那个才会移动,如果这个指针对应的位置全局最高,这个指针不会动。
-
上面代码重复的地方在于,每次对于一个p柱,都会计算左边最高的,右边最高的。现在我们用双指针,leftMax记录左指针走到现在最高的地方,rightMax记录右指针走到现在最高的地方。
-
使用双指针l,r 每次移动l和r中高度较小的那个(下称p), 较小的那个将会承担起计算上部雨水的职责。
-
比较p的高度和这侧的max高度:
如果p高,那p柱上边不会有雨水,更新那侧的最大值;
如果p低,计算p柱上边的雨水(当侧max-p对应的height)
可不可能出现下图的情况呢?
此时,r当侧maxHeight为5,5 - 2 = 3,但实际2这里只有4-2的水
这种情况是不可能的!因为r指针如果会比左面的wall高,那么r指针根本不会移动,像上图的情况,2之后会对应l指针 -
注意⚠️:如果我移动了左边的指针left,只可能是因为左边的指针之前遍历到的位置都比右边的小
class Solution {
public int trap(int[] height) {
int leftMax = 0, rigthMax = 0;
int l = 0, r = height.length - 1;
int sum = 0;
while(l < r){
if (height[l] <= height[r]){
if (height[l] < leftMax){
sum += leftMax - height[l];
}else {
leftMax = height[l];
}
l++;
}else {
if (height[r] < rigthMax){
sum += rigthMax - height[r];
}else {
rigthMax = height[r];
}
r--;
}
}
return sum;
}
}