https://leetcode-cn.com/problems/trapping-rain-water/
给定 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
能接多少雨水,可以先单独计算每列能否接水,则分别计算左边最高的柱子
left[i] = max(left[i - 1], height[i - 1]);
和右边最高的柱子
right[i] = max(right[i + 1], height[i + 1]);
当前列能接的水,由左右两边较高度较低的柱子决定。较小的列大于当前列,则能接到雨水,否则接不到雨水。
#define min(a, b) (a < b ? a : b)
#define max(a, b) (a > b ? a : b)
int trap(int* height, int heightSize){
int sum = 0;
int *left = (int*)calloc(heightSize, sizeof(int));
int *right = (int *)calloc(heightSize, sizeof(int));
//从[1] 到 [lenght-2],最两端情况不用考虑
for (int i = 1; i < heightSize; i++) {
left[i] = max(left[i - 1], height[i - 1]); //找出左边最高列
}
for (int i = heightSize -2 ; i >= 0; i--) {
right[i] = max(right[i + 1], height[i + 1]); //找出右边最高列
}
for (int i = 0; i < heightSize; i++) {
int level = min(left[i], right[i]); //找出两端较小的
sum += max(0, level - height[i]); //较小的列大于当前列,有雨水,否则是0
}
return sum;
}