【leetcode】Trapping Rain Water(hard)

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. 

思路：

我自己的想法，先遍历一遍数组找到所有的山峰位置，在例子中是1,3,7,10，然后再填充两个山峰之间的水，获得水量

class Solution {
public:
    int trap(int A[], int n) {
        bool haswater = false;
        vector<int> peak;
        int ans = 0;
        for(int i = 0; i < n - 1; i++) //找到第一个山峰
        {
            if(A[i] > A[i + 1])
            {
                peak.push_back(i);
                break;
            }
        }
        if(peak.empty())
        {
            return 0; //没有山谷 没水
        }
        int maxheight = A[peak[0]];
        for(int i = peak[0] + 1; i < n; i++) //找剩下的山峰
        {
            if(A[i] > A[i - 1]) //比上一个值大 新山峰
            {
                while(A[i] > A[peak.back()] && A[peak.back()] < maxheight) //比之前的山峰高，且之前的山峰不是最高峰，删掉之前的 第一个山峰不能删
                {
                    peak.pop_back();
                }
                maxheight = (A[i] > maxheight) ? A[i] : maxheight; //新的最高峰
                peak.push_back(i); 
            }
        }

        for(int i = 0; i < peak.size() - 1; i++) //获得水量 依次向两个山峰间加水
        {
            int height = min(A[peak[i]], A[peak[i+1]]); //水平面是两个山峰中矮一点的那个
            for(int j = peak[i] + 1; j < peak[i+1]; j++)
            {
                ans += max(height - A[j], 0); //防止5 4 1 2 这种水平面比中间非山峰处低的情况
            }
        }

        return ans;
    }
};

大神的思路：没有像我一样先找所有的山峰，而是边遍历边获得每个坐标位置的水量。 而且大神是左右两边收缩的。

class Solution {
public:
    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;
    }
};