LeetCode--42

42. 接雨水

给定 n 个非负整数表示每个宽度为 1 的柱子的高度图，计算按此排列的柱子，下雨之后能接多少雨水。

示例 1：

输入：height = [0,1,0,2,1,0,1,3,2,1,2,1]
输出：6
解释：上面是由数组 [0,1,0,2,1,0,1,3,2,1,2,1] 表示的高度图，在这种情况下，可以接 6 个单位的雨水（蓝色部分表示雨水）。 

示例 2：

输入：height = [4,2,0,3,2,5]
输出：9

提示：

• n == height.length
• 1 <= n <= 2 * 104
• 0 <= height[i] <= 105

先看我的代码：

class Solution {
public:
    int trap(vector<int>& height) {
        int len=height.size();
        auto Maxheight=max_element(height.begin(),height.end());
        int maxheight=*Maxheight;
        int num=0;
        for(int i=1;i<=maxheight;i++)
        {
            vector<int> a;
            int id=-1;
            for(int j=0;j<len;j++)
            {
                if(height[j]>=i)
                {
                a.push_back(j);
                id++;
                if(id>0)
                num+=a[id]-a[id-1]-1;
                }

            }
        }
        return num;
    }
};

有一个小缺陷，不能通过所有的测试案例：

只有4个测试案例没有通过，说明我的思路是正确的，但是没有考虑到时间复杂度，说没有考虑也不对，我刚开始写出的代码是这样的：

class Solution {
public:
    int trap(vector<int>& height) {
        int len=height.size();
        auto Maxheight=max_element(height.begin(),height.end());
        int maxheight=*Maxheight;
        int num=0;
        for(int i=1;i<=maxheight;i++)
        {
            vector<int> a;
            for(int j=0;j<len;j++)
            {
                if(height[j]>=i)
                a.push_back(j);

            }
            int Si=a.size();
            if(Si>=2)
            {
                for(int h=0;h<Si-1;h++)
                {
                    num+=a[h+1]-a[h]-1;
                }
            }
        }
        return num;
    }
};

显然，这样时间复杂度更大。

下面 是标准答案：

class Solution {
public:
    int trap(vector<int>& height) {
        int n=height.size();
        if(n==0)
        {
            return 0;
        }
        vector<int>leftMax(n);
        leftMax[0]=height[0];
        for(int i=1;i<n;i++)
        {
            leftMax[i]=max(leftMax[i-1],height[i]);
        }
        vector<int>rightMax(n);
        rightMax[n-1]=height[n-1];
        for(int i=n-2;i>=0;i--)
        {
            rightMax[i]=max(rightMax[i+1],height[i]);
        }
        int ans=0;
        for(int i=0;i<n;i++)
        {
            ans+=min(leftMax[i],rightMax[i])-height[i];
        }
        return ans;

    }
};