Trapping Rain Water Problem

1
2
3
Input: height = [0,1,0,2,1,0,1,3,2,1,2,1]
Output: 6
Explanation: The above elevation map (black section) 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.

1
2
Input: height = [4,2,0,3,2,5]
Output: 9

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
class Solution {
public:
    /*  Instead of calculating area by height*width, we can think it in a cumulative way. 
        In other words, sum water amount of each bin(width=1).
        Search from left to right and maintain a max height of left and right separately, 
        which is like a one-side wall of partial container. Fix the higher one and flow water from the lower part. 
        For example, if current height of left is lower, we fill water in the left bin. Until left meets right, 
        we filled the whole container. */
    int trap(int A[], int n) {
        int left=0; int right=n-1;
        int res=0;
        int maxleft=0, maxright=0;
        while (left<=right){
            if (A[left]<=A[right]) {
                if (A[left]>=maxleft) maxleft=A[left];
                else res+=maxleft-A[left];
                left++;
            }
            else {
                if (A[right]>=maxright) maxright= A[right];
                else res+=maxright-A[right];
                right--;
            }
        }
        return res;
    }
};