[LeetCode]Trapping Rain Water

来源:互联网 发布:tensorflow中文手册 编辑:程序博客网 时间:2024/06/13 06:18

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!

参考:戴方勤 (soulmachine@gmail.com)     https://github.com/soulmachine/leetcode

public class Solution {    public int trap(int[] A) {        int sum = 0;    int maxLeft [] = new int[A.length];        int maxRight [] = new int [A.length];        for(int i=1;i<A.length;i++){        maxLeft[i] = Math.max(maxLeft[i-1], A[i-1]);        maxRight[A.length-1-i] = Math.max(maxRight[A.length-i], A[A.length-i]);        }        for(int i=0;i<A.length;i++){        int height = Math.min(maxLeft[i],maxRight[i]);        if(height>=A[i]){        sum += height-A[i];        }        }        return sum;    }}





0 0
原创粉丝点击