[leetcode]42. Trapping Rain Water雨水积水问题

Given n non-negative integers representing an elevation map where the width of each bar is 1, compute how much water it is able to trap after raining.

The above elevation map is represented by array [0,1,0,2,1,0,1,3,2,1,2,1]. In this case, 6 units of rain water (blue section) are being trapped. Thanks Marcos for contributing this image!

 

Example:

Input: [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6

 

题意：

给定一个地形，计算能存多少雨水。

 

思路：

1. 扫数组，找到最高柱子并以此为中心，将数组分为两半。 

2. 对左半部分而言，可以想象右边一定有堵墙，只需要每次更新leftMost就能决定water大小

3. 对右半部分而言，同理。

 

代码：

class Solution {
    public int trap(int[] height) {
        // find hightest bar 
        int peak_index = 0; 
        for(int i = 0; i < height.length; i++){
            if(height[i] > height[peak_index]){
                peak_index = i; 
            }
        } 
        // 1. from left to peak_index
        int leftMostBar = 0; 
        int water = 0; 
        for(int i = 0; i < peak_index; i++){
            if (height[i] > leftMostBar){
                leftMostBar = height[i];
            }else{
                water  = water + leftMostBar - height[i]; 
            }
        }
        
        // 2. from right to peak_index
        int rightMostBar = 0; 
        for(int i = height.length - 1; i > peak_index; i--){
            if (height[i] > rightMostBar){
                rightMostBar = height[i];
            }else{
                water  = water + rightMostBar - height[i]; 
            }
        }
        return water;        
    }
}