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

LeetCode 042 Trapping Rain Water

时间:2015-02-07 21:33:12      阅读:145      评论:0      收藏:0      [点我收藏+]

标签:

题目要求: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.

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!

 

代码如下:

class Solution {
public:
    int trap(int A[], int n) {
        
        int maxIdx = 0;
        int water = 0;
        
        //找到最长的木板,设为maxIdx
        for(int i = 1; i < n; i++){
            if(A[i] > A[maxIdx]){
                maxIdx = i;
            }
        }
        
        int max = A[0];
        
        //左侧逼近
        for(int i = 1; i < maxIdx; i++){
            
            if(max < A[i]) 
                max = A[i];
            //木板左边(max)和右边(最高)都比它高,则可以放
            // max - 该木板长度 的水
            else 
                water += max - A[i];
        }
        
        //右侧逼近
        max = A[n - 1];
        for(int i = n - 2; i >= maxIdx; i--){
            if(max < A[i]) max = A[i];
            else water += max - A[i];
        }
        
        return water;
        
    }
};

 

LeetCode 042 Trapping Rain Water

标签:

原文地址:http://www.cnblogs.com/510602159-Yano/p/4279257.html

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