码迷,mamicode.com
首页 > 移动开发 > 详细

LeetCode - Trapping Rain Water

时间:2015-12-25 13:30:41      阅读:144      评论: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!

思路:

1. 找到最高的点index

2. 计算(0, index)和(index, n-1)之间各点的面积

package area;

public class TrappingRainWater {

    public int trap(int[] height) {
        int n;
        if (height == null || (n = height.length) == 0) return 0;
        int area = 0;
        int maxHeight = height[0];
        int maxHeightIndex = 0;
        for (int i = 1; i < n; ++i) {
            if (height[i] > maxHeight) {
                maxHeight = height[i];
                maxHeightIndex = i;
            }
        }
        
        int leftMax = height[0];
        for (int i = 1; i < maxHeightIndex; ++i) {
            if (height[i] > leftMax)
                leftMax = height[i];
            area += (leftMax - height[i]);
        }
        
        int rightMax = height[n - 1];
        for (int i = n - 2; i > maxHeightIndex; --i) {
            if (height[i] > rightMax) 
                rightMax = height[i];
            area += rightMax - height[i];
        }
        
        return area;
    }
    
    public static void main(String[] args) {
        // TODO Auto-generated method stub
        int[] height = { /*0,1,0,2,1,0,1,3,2,1,2,1*/ 5,2,1,2,1,5 };
        TrappingRainWater t = new TrappingRainWater();
        System.out.println(t.trap(height));
    }

}

 

LeetCode - Trapping Rain Water

标签:

原文地址:http://www.cnblogs.com/shuaiwhu/p/5075536.html

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