题目传送门:42. 接雨水 - 力扣(LeetCode)
题解
题目大意是给定一排柱子的高度,要求计算柱子中能存多少雨水,如下图所示。
一道思路题,让人不禁联想到当年noip花匠。经过思考后可以发现每个位置能接到的雨水量取决于该位置左侧的最大高度left,该位置右侧的最大高度right以及当前位置的高度height,其实是left和right中的较小值减去当前位置的高度,即min(left,right)-height。比如在图中第二列,左侧最大高度为1,右侧最大高度为3,自身高度为0,那么该处能存的水量取决于较小一侧的最大高度减自身高度,即1-0=1。又比如图中第三列,左侧最大高度为2(算自己),右侧最大高度为3,自身高度为2,那么该处存水量为min(2,3)-2=0。
在代码中,我们通过一遍从前向后和一遍从后向前的循环在left[i]中记录位置i左侧最大高度,right[i]中记录位置i右侧最大高度,最后通过公示min(left[i],right[i])-height[i]计算每个位置的存水量。实际上,在从后向前的循环中就已经可以记录每个位置的存水量了。
参考代码
class Solution {
public:
int trap(vector<int>& height) {
int n=height.size(),maxhigh,ans=0;
int left[20005],right[20005];
maxhigh=0;
for(int i=0;i<n;i++)
{
if(height[i]>maxhigh)maxhigh=height[i];
left[i]=maxhigh;
}
maxhigh=0;
for(int i=n-1;i>=0;i--)
{
if(height[i]>maxhigh)maxhigh=height[i];
right[i]=maxhigh;
}
for(int i=0;i<n;i++)
ans+=min(left[i],right[i])-height[i];
return ans;
}
};