首页 > 代码库 > LeetCode:Trapping Rain Water
LeetCode: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
.
思路:从左到右依次扫描各列,找出当前列左边的最大高度hli和右边的最大高度hri。如果hli>A[i]并且hri>A[i],则当前列所能盛水的体积为min(hli,hri)- A[i]。
代码:
int Solution::trap(int A[],int n) { int i,j; int count = 0; int * left_height_array = (int *)malloc(sizeof(int)*n); int left_height_max = 0; for(i = 0;i < n;i++) { left_height_array[i] = left_height_max; if(A[i] > left_height_max) left_height_max = A[i]; } int * right_height_array = (int*)malloc(sizeof(int)*n); int right_height_max = 0; for(i = n-1;i >=0;i--) { right_height_array[i] = right_height_max; if(A[i] > right_height_max) right_height_max = A[i]; } for(i = 1;i < n-1;i++) { int left_height = left_height_array[i]; int right_height = right_height_array[i]; if(left_height > A[i] && right_height > A[i]) { if(left_height > right_height) count = count + (right_height - A[i]); else count = count + (left_height - A[i]); } } return count; }
LeetCode:Trapping Rain Water
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。