题目链接: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.
^
3| ■ □: water
2| ■ □ □ □ ■ ■ □ ■ ■: elevation map
1| ■ □ ■ ■ □ ■ ■ ■ ■ ■ ■
————————————————————————>
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.
这道题的要求是计算最多能装多少水。其中,数组中的数字表示高度。
这道题的思路是采用l和r两个指针,维护装水两边的位置。
当l处高度低时,说明l右侧装的水肯定和l处一样高,此时逐步右移l,同是加上l处与右移后位置高度差(因为这里都能装水啊),直到再遇到同样高或者更高的位置。然后进行下一轮判断。
同样,当r处高度低时,说明r左侧的水肯定和r处一样高,此时逐步左移r,同是加上r处与左移后位置高度差,直到再遇到同样高或者更高的位置。
最后直到l和r相遇,结束。
时间复杂度:O(n)
空间复杂度:O(1)
1 class Solution
2 {
3 public:
4 int trap(int A[], int n)
5 {
6 int res = 0, l = 0, r = n - 1;
7
8 while(l < r)
9 {
10 int minh = min(A[l], A[r]);
11 if(A[l] == minh)
12 while(++ l < r && A[l] < minh)
13 res += minh - A[l];
14 else
15 while(l < -- r && A[r] < minh)
16 res += minh - A[r];
17 }
18
19 return res;
20 }
21 };