思路:
当前柱子 i 能接受的雨水为左边最高的柱子l_max,右边最高的柱子r_max中较矮的一个min, 如果min > 当前柱子的高度height[i], 则当前柱子能接收的水为 min - height[i]
方法1:暴力法,左右维护一个最高柱子
public int trap(int[] height) {
int n = height.length;
int ans = 0;
// 第一个柱子和最后一个柱子不会有雨水,所以 i -> [1,n-2]
for (int i = 1; i < n - 1; i++) {
int l_max = 0, r_max = 0;
// 维护左右两边的最大值,注意边界条件
for (int j = i; j >= 0; j--) {
l_max = Math.max(l_max, height[j]);
}
for (int j = i; j < n; j++) {
r_max = Math.max(r_max, height[j]);
}
//找出两端较小的
int min = Math.min(l_max, r_max);
//只有较小的一段大于当前列的高度才会有水,其他情况不会有水
if (min > height[i]) {
ans += min - height[i];
}
}
return ans;
}
方法2:动态规划,我们可以用数组来存储每个位置左边最高的柱子和右边最高的柱子
l_max[i] = Math.max(l_max[i - 1], height[i - 1]);
r_max[i] = Math.max(r_max[i + 1], height[i + 1]);
public int trap2(int[] height) {
int n = height.length;
int[] l_max = new int[n];
int[] r_max = new int[n];
int ans = 0;
for (int i = 1; i < n; i++) {
l_max[i] = Math.max(l_max[i - 1], height[i - 1]);
}
for (int i = n - 2; i >= 0; i--) {
r_max[i] = Math.max(r_max[i + 1], height[i + 1]);
}
// 第一个柱子和最后一个柱子不会有雨水,所以 i -> [1,n-2]
for (int i = 1; i < n - 1; i++) {
int min = Math.min(l_max[i], r_max[i] );
if (min > height[i]){
ans += min - height[i];
}
}
return ans;
}
方法3:动态规划一般都可以优化空间复杂度,双指针
只要 r_max[i] > l_max[i] (元素 0 到元素 6),积水高度将由 l_max 决定
public int trap3(int[] height) {
int left = 0, right = height.length - 1;
int ans = 0;
int left_max = 0, right_max = 0;
while (left < right) {
if (height[left] < height[right]) {
if (height[left] >= left_max) {
left_max = height[left];
} else {
ans += (left_max - height[left]);
}
++left;
} else {
if (height[right] >= right_max) {
right_max = height[right];
} else {
ans += (right_max - height[right]);
}
--right;
}
}
return ans;
}