码迷,mamicode.com
首页 > 其他好文 > 详细

*Trap Rain Water

时间:2015-10-30 07:03:53      阅读:265      评论:0      收藏:0      [点我收藏+]

标签:

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.

For example, 
Given [0,1,0,2,1,0,1,3,2,1,2,1], return 6.

技术分享

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!

 

public class Solution {
    public int trap(int[] height) 
    {
        int len = height.length;
        if(len<=2)return 0;
        int maxIndex = -1;
        int maxHgt = Integer.MIN_VALUE;
        for (int i=0;i<len;i++)
        {
            if(height[i]>maxHgt)
            {
                maxHgt = height[i];
                maxIndex = i;
            }
        }
        
        int leftMax = height[0];
        int rightMax = height[len-1];
        int water = 0;
        
        for (int i=1;i<maxIndex;i++)
        {
            if(height[i]<leftMax)
            {
                water += leftMax - height [i]; 
            }
            else
            {
                leftMax = height[i];
            }
            
        }
        
        for(int i=len-2;i>maxIndex;i--)
        {
            if(height[i]<rightMax)
            {
                water += rightMax - height [i]; 
            }
            else
            {
                rightMax = height[i];
            }
            
        }
        
        return water;
        
    }
}

 

*Trap Rain Water

标签:

原文地址:http://www.cnblogs.com/hygeia/p/4922269.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!