//按列计算,按行计算超出时间限制了
//每一列能装的水,取决于左右两边的最高列的高度,如果有一边最高列比当前列矮,则不能装水,如果两边都有更高的,则装水的量等于比较矮的那一列的高度减去当前列的高度
class Solution {
public int trap(int[] height) {
int sum = 0;
//保存着左边和右边最高列的坐标
int max_left = 0, max_right = 0;
//最两端的列不用考虑,因为一定不会有水。所以下标从 1 到 length - 2
int left = 1;
int right = height.length - 2; // 加右指针进去
for (int i = 1; i < height.length - 1; i++) {
//如果左边的更矮,则找左边最高的列
if (height[left - 1] < height[right + 1]) {
//求左边的最高列
max_left = Math.max(max_left, height[left - 1]);
int min = max_left;
//如果矮的列比当前列高,当前列才能装水
if (min > height[left]) {
sum = sum + (min - height[left]);
}
left++;
//同理求右边的
} else {
max_right = Math.max(max_right, height[right + 1]);
int min = max_right;
if (min > height[right]) {
sum = sum + (min - height[right]);
}
right--;
}
}
return sum;
}
}
leetcode42. 接雨水
最新推荐文章于 2024-10-07 11:53:29 发布