leetcode(42) - 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个单位的雨水（蓝色部分）被装。

对于每个柱子，找到其左右两边最高的柱子，该柱子能容纳的面积就是 min(leftMostHeight,rightMostHeight) - height。所以，

1. 从左往右扫描一遍，对于每个柱子，求取左边最大值；

2. 从右往左扫描一遍，对于每个柱子，求最大右值；

3. 再扫描一遍，把每个柱子的面积并累加。

int trap(int* height, int heightSize) {
    
    if(heightSize <= 2) return 0;
    
    int i = 0;
    int* left_max=malloc(sizeof(int) * heightSize);
    memset(left_max, 0, heightSize);
    int* right_max=malloc(sizeof(int) * heightSize);
    memset(right_max, 0, heightSize);
    
    int left_tmp_max=0, right_tmp_max=0, ret=0;
    
    while(i < heightSize){    // 针对每个元素i，寻找它左边最大的，保存在left_max[i]
        left_max[i] = left_tmp_max;
        if(left_tmp_max < height[i])
            left_tmp_max=height[i];
        i++;
    }

    i=heightSize-1;
    while(i >= 0){             // 针对每个元素i， 寻找它右边最大的，保存在right_max[i]
        right_max[i] = right_tmp_max;
        if(right_tmp_max < height[i])
            right_tmp_max=height[i];
        i--;
      
       
    }
    
    i=0;
    while(i<heightSize){        //处理每个元素i，将min(left_max[i], right_max[i])-height[i] 加到ret中，即每个元素上面可以存储的水
        if(left_max[i] > right_max[i]) {
            ret+= ((right_max[i]-height[i])>0 ? (right_max[i]-height[i]) : 0);
        } else {
            ret+= ((left_max[i]-height[i])>0 ? (left_max[i]-height[i]) : 0);
        }
        i++;
         
    }
    
    return ret;
    
}