LeetCode 42 接雨水

  • 1.题目链接
https://leetcode-cn.com/problems/trapping-rain-water/
  • 2.题目描述

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图,计算按此排列的柱子,下雨之后能接多少雨水。

LeetCode 42 接雨水_第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.提交记录

LeetCode 42 接雨水_第2张图片

你可能感兴趣的:(LeetCode 42 接雨水)