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