leetcode面试经典150题-接雨水

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

示例 1：

输入：height = [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 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

输入：height = [4,2,0,3,2,5]
输出：9

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

int trap(int* height, int heightSize){
    int max=0,max_index=0;
    int water_lvl=0,water=0;
    // 找到水位最高点
    for(int i=0;i<heightSize;i++){
        if(height[i]>max){
            max=height[i];
            max_index=i;
        }
    }
    // 从左向最高点计算水量
    water_lvl=0;
    for(int i=0;i<max_index;i++){
        if(height[i]>water_lvl){
            water_lvl=height[i];
        }
        water+=water_lvl-height[i];
    }
    // 同理 从右向最高点计算水量
    water_lvl=0;
    for(int i=heightSize-1;i>max_index;i--){
        if(height[i]>water_lvl){
            water_lvl=height[i];
        }
        water+=water_lvl-height[i];
    }
    return water;
}