LeetCode 42 接雨水

• 1.题目链接

https://leetcode-cn.com/problems/trapping-rain-water/

• 2.题目描述

给定 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

• 3.题目解析

能接多少雨水，可以先单独计算每列能否接水，则分别计算左边最高的柱子

left[i] = max(left[i - 1], height[i - 1]);

和右边最高的柱子

right[i] = max(right[i + 1], height[i + 1]);

当前列能接的水，由左右两边较高度较低的柱子决定。较小的列大于当前列，则能接到雨水，否则接不到雨水。

• 4.代码实现

#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;
}

• 5.提交记录