此题不难,但要注意:
1)与最大直方图面积问题区分: 如果输入是a[3]={2,1,2}, 直方图面积会是3=1*3,而该蓄水池问题的答案是4=2*2。因为直方图问题主要考虑区间中最低的那个,而该蓄水池问题会忽略掉中间低的部分。
2)不需要考虑挡板,也就是说a[3]={1,2,1}的结果是2。
解法为用两个指针,一前一后扫描。如果前面的高,则移后面的指针,反之移前面的指针。然后每次都记录两指针值之间的面积,最后返回最大值即可。
int maxArea(vector<int>& height) {
int p1=0, p2=height.size()-1;
int volume=0, maxVolume=0;
while(p1<p2) {
int volume;
if (height[p2]>height[p1]) {
volume=(p2-p1)*height[p1];
p1++;
}else{
volume=(p2-p1)*height[p2];
p2--;
}
maxVolume=max(maxVolume, volume);
}
return maxVolume;
}