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

题目描述

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

上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 感谢 Marcos 贡献此图。

示例:

输入: [0,1,0,2,1,0,1,3,2,1,2,1]
输出: 6

解题思路

找到最高的那个柱子，把数组分成两部分，对于两部分都已经确定了一个边界高度了，所以对剩余的每个柱子至于确定一边的边界高度值，就可以直接计算出能接的雨水了。

int trap(int* height, int heightSize) {
    int h, t = 0, ans = 0, max = height[0];
    for (int i = 1; i < heightSize; i++) {
        if (height[i] > max) {
            max = height[i];
            t = i;
        }
    }
    h = height[0];
    for (int i = 1; i < t; i++) {
        if (h > height[i])
            ans += h - height[i];
        else  h = height[i];
    }
    h = height[heightSize - 1];
    for (int i = heightSize - 2; i > t; i--) {
        if (h > height[i])
            ans += h - height[i];
        else  h = height[i];
    }
    return ans;
}