leetcode算法题42. Trapping Rain Water大神代码[转]及个人思路（c++）

leetcode算法题42. Trapping Rain Water大神代码[转]及个人思路（c++）

原题：

大神代码

转自https://leetcode.com/problems/trapping-rain-water/discuss/17364/7-lines-C-C%2B%2B
代码：(来自StefanPochmann)

int trap(vector<int>& height) {
    int l = 0, r = height.size()-1, level = 0, water = 0;
    while (l < r) {
        int lower = height[height[l] < height[r] ? l++ : r--];
        level = max(level, lower);
        water += level - lower;
    }
    return water;
}

大神代码分析及思路：

定义总蓄水量 water，当前位置的水位高度 level 和当前位置地面高度 lower
将 l 和 r 分别作为左右两侧的起始位置，同时向中间移动
只要 l 和 r 没有相等，就是没有到最高点。
每次判断 height[ l ] 和 height[ r ] 的大小，小的方向向中间靠近 1 格，
同时更新 level，也就是如果 lower 比 level 大，说明现在在 l 和 r 之间的格子水位至少有 lower 的高度，因为这些格子的左墙壁是height[ l ] ,右墙壁是 height[ r ]，而 lower 是其中较小的那个。
最后更新总蓄水量 water ，也就是当前水位减去当前地面高度的差值。

个人解法

思路：

首先是找到中间的最高点（可能不止一个）
新建一个数组 heightfull ，作为蓄满水之后各个位置的高度。
找最高点的时候就确定最高点中最左侧和最右侧的位置。
将两个最高点中间的部分设置为最高点高度。（因为左右两侧的墙高度是这个）
之后分别从左侧和右侧开始，每次从当前的最高点向侧面寻找下一个最靠近侧面的次高点，同时将次高点到最高点中间用次高点的高度填满，一直重复直到新的点最高值不大于0。
最后计算数组 heightfull 和最初的数组 height 之间的差值，既是蓄水量。
代码：

class Solution {
public:
    int trap(vector<int>& height) {
        vector<int> heightfull(height);
        int length=height.size();
        int max = 0;
        int maxidleft = -1;
        int maxidright = -1;
        int leftid = -1;
        int rightid = -1;
        int i;
        int sum=0;
        if(length<3)
            return 0;
        for (i=0;i<length;i++)
        {
            if(max<height[i])
            {
                max = height[i];
                maxidright = maxidleft = i;
            }
            else if(max == height[i])
            {
                maxidright = i;
            }
        }
        leftid = maxidleft;
        rightid = maxidright;
        for(i=maxidleft;i<=maxidright;i++)
        {
            heightfull[i] = max; 
        }
        while(true)
        {
            max = 0;
            for(i=rightid+1;i<length;i++)
            {
                if(max<=height[i])
                {
                    max = height[i];
                    maxidright = i;
                }
            }
            if(max>0)
            {
                for(i=rightid+1;i<maxidright;i++)
                {
                    heightfull[i] = max; 
                }
            }
            else
                break;
            rightid = maxidright;
        }
        while(true)
        {
            max = 0;
            for(i=leftid-1;i>=0;i--)
            {
                if(max<=height[i])
                {
                    max = height[i];
                    maxidleft = i;
                }
            }
            if(max>0)
            {
                for(i=leftid-1;i>maxidleft;i--)
                {
                    heightfull[i] = max; 
                }
            }
            else
                break;
            leftid = maxidleft;
        }
    
        for (i=0;i<length;i++)
        {
            sum+=heightfull[i]-height[i];
        }
        return sum;
    }
    

};