给定n个非负整数代表每个柱的宽度的高程图1，计算下雨后可以捕集多少水。...

最新推荐文章于 2021-08-25 22:46:50 发布

weixin_34303897

最新推荐文章于 2021-08-25 22:46:50 发布

阅读量704

点赞数

文章标签： python

原文链接：https://my.oschina.net/u/586094/blog/1620699

版权

2019独角兽企业重金招聘Python工程师标准>>>

/*
* @param heights: a list of integers
* @return: a integer
*/
public static int trapRainWater(int[] heights) {
// write your code here
   if(heights ==null || heights.length<2) return 0;
   int num =0;
   int length = heights.length;
   int[] leftArray = new int[length];
   int leftMax = heights[0];
   leftArray[0]=0;
   int[] rightArray = new int[length];
   rightArray[length-1]=0;
   for(int i=1;i<length;i++) {
       leftArray[i]=leftMax;
       if(leftMax<heights[i]) {
       leftMax = heights[i];
       }
   }
   int rightMax = heights[length-1];
   for(int i=length-2;i>=0;i--) {
       rightArray[i]=rightMax;
       int numtmp = Math.min(rightArray[i], leftArray[i])-heights[i];
       if(numtmp>0) num +=numtmp;
       if(rightMax<heights[i]){
           rightMax=heights[i];
       }
   }

   return num;

}