class Solution {
public int trap(int[] height) {
int n = height.length - 1;
int sum = 0;
//定义两个数组,分别用来存放当前位置,左右两边最高的柱子。
int[] max_left = new int[n];
int[] max_right = new int[n];
for (int i = 1; i < n; i++) {
//每次遍历都找到i左边最高的那一跟柱子
//比如i++了意味着往右移动了一格到了一根新的柱子,也同时意味着上一根i所在的柱子成为了新增的柱子需要与max_left进行比较,
// 则当前max_left[i]更新为上一轮的max_left[i-1]与新增的height[i-1]比较之后的最大值,也即得到当前i左边最高的柱子。
max_left[i] = Math.max(max_left[i - 1], height[i - 1]);
}
for (int i = n - 1; i >= 0; i--) {
//同理找右边最高的柱子
max_right[i] = Math.max(max_right[i + 1], height[i + 1]);
}
for (int i = 1; i < n; i++) {
//遍历每一根柱子,开始计算每一根柱子能接多少雨水
//第0根柱子左边没有柱子,第n根柱子右边没有柱子,因此不能存放雨水,故循环从1到n
int min = Math.min(max_right[i], max_left[i]); //获取当前柱子左右两侧。最高柱子中矮的一侧,雨水存放高度由矮的一侧决定
if (min > height[i]) { //当前柱子高度小于两侧最大高度的比较小值,该柱子才 能存放雨水
sum += min - height[i]; //将每一根柱子能存放的雨水加起来就可以得到最终的总量
}
}
return sum;
}
}
力扣42.接雨水 动态规划解法
于 2023-11-28 12:56:48 首次发布