Trapping Rain Water (Bar Height) -- 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!

思路:使用两个指针left和right,分别从两端向中间靠拢。同时,记录left和right曾经到过的最高值。如果当前值并没有超过最高值,则将最高值与当前的差值作为水的量,并将指针加一(right指针是减一)。为了让这个方法有效(无效的情况是,指针后续遇见的bar都不会再高于最高值了,那么这些水根本存不起来),因此我们每次只移动left和right中矮的那一个(如果两者相等,则移动left)。这样,一定能保证最后会碰到比最高值要高的bar(至少是等高),因为我们移动的是矮的那个指针,往前移动肯定最后会遇见另一个高的指针。

 1 class Solution {
 2 public:
 3     int trap(vector<int>& height) {
 4         if (height.size() == 0) return 0;
 5         int left = 0, right = height.size() - 1, res = 0;
 6         int maxleft = height[left], maxright = height[right];
 7         while (left < right)
 8         {
 9             if (height[left] <= height[right])
10             {
11                 if (height[left] > maxleft) maxleft = height[left];
12                 else res += maxleft - height[left];
13                 left++;
14             }
15             else
16             {
17                 if (height[right] > maxright) maxright = height[right];
18                 else res += maxright - height[right];
19                 right--;
20             }
21         }
22         return res;
23     }
24 };

 

bar height问题:这是Amazon面试中的一道问题。求最高的液面高度。这里只需要把上方的代码修改一下就可以用。就是在res值变更时,若max与当前高度差值非零时,记录下max值,最后最高的max值就是结果。

 1 class Solution {
 2 public:
 3     int trap(vector<int>& height) {
 4         if (height.size() == 0) return 0;
 5         int left = 0, right = height.size() - 1, res = 0;
 6         int maxleft = height[left], maxright = height[right];
 7         while (left < right)
 8         {
 9             if (height[left] <= height[right])
10             {
11                 if (height[left] > maxleft) maxleft = height[left];
12                 else if (maxleft - height[left] > 0)
13                     res = max(res, maxleft);
14                 left++;
15             }
16             else
17             {
18                 if (height[right] > maxright) maxright = height[right];
19                 else if (maxright - height[right] > 0)
20                     res = max(res, maxright);
21                 right--;
22             }
23         }
24         return res;
25     }
26 };

 

posted @ 2016-01-30 05:37  fenshen371  阅读(169)  评论(0编辑  收藏  举报