首页 > 代码库 > 【leetcode】Trapping Rain Water
【leetcode】Trapping Rain Water
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!
找到最长的木板所在的位置，分别从左侧和右侧向其逼近。
在左侧逼近过程中：
如果一个木板的长度小于已经遍历的最大长度max，即max>该木板<maxIdx，所以在该木板位置能存max - 该木板长度的水量（左右两侧各有一个木板长于它）。
如果一个木板的长度大于已经遍历的最大长度max，即max<该木板<maxIdx，所以在该木板位置不能存水（因为左右两侧只有一个木板（maxIdx）长于它）。更新max值。
右侧逼近过程与左侧相似。
1 class Solution { 2 public: 3 int trap(int A[], int n) { 4 5 if(n==0) return 0; 6 7 int i; 8 int maxIndex=0; 9 int Amax=A[0];10 11 for(i=1;i<n;i++)12 {13 if(Amax<A[i]) 14 {15 Amax=A[i];16 maxIndex=i;17 }18 }19 20 int curMax=A[0];21 int totalWater=0;22 23 for(i=1;i<maxIndex;i++)24 {25 if(A[i]<curMax) totalWater+=curMax-A[i]; 26 else curMax=A[i];27 }28 29 curMax=A[n-1];30 for(i=n-2;i>maxIndex;i--)31 {32 if(A[i]<curMax) totalWater+=curMax-A[i];33 else curMax=A[i];34 }35 36 return totalWater;37 }38 };
【leetcode】Trapping Rain Water
声明：以上内容来自用户投稿及互联网公开渠道收集整理发布，本网站不拥有所有权，未作人工编辑处理，也不承担相关法律责任，若内容有误或涉及侵权可进行投诉： 投诉/举报 工作人员会在5个工作日内联系你，一经查实，本站将立刻删除涉嫌侵权内容。