首页 > 代码库 > Trapping Rain Water -- leetcode
Trapping Rain Water -- leetcode
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!
左边右边中短者为堤坝。比堤坝更短者可蓄一定的水。逐步收集这些水量。比当前堤坝更高者,将成为新堤坝。
此算法在leetcode上实际执行时间为10ms。
class Solution { public: int trap(int A[], int n) { int low = 0, high = n-1; int rain = 0, bar = 0; while (low < high) { if (A[low] < A[high]) { if (bar < A[low]) bar = A[low]; else rain += bar-A[low]; ++low; } else { if (bar < A[high]) bar = A[high]; else rain += bar-A[high]; --high; } } return rain; } };
方法二,假定都是雨水,再逐步刨出那些堤坝占据的水量。
此算法在leetcode 上实际执行时间为13ms。比上面慢点,可能是因为用到了乘法。
class Solution { public: int trap(int A[], int n) { int low = 0, high = n-1; int bar = 0, rain = 0; while (low < high) { const int newbar = min(A[low], A[high]); if (newbar > bar) { rain += (newbar-bar) * (high-low); bar = newbar; } if (A[low] < A[high]) { rain -= min(bar, A[low]); ++low; } else { rain -= min(bar, A[high]); --high; } } return rain; } };
Trapping Rain Water -- leetcode
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。