【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;

    }

};

 

你可能感兴趣的:(LeetCode)