给你 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0) 。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
题目来源:LeetCode11. 盛最多水的容器
思路
双指针:将指针l
和r
分别置于最左和最右,将两指针视为容器的左壁和右壁计算答案并更新最大值,每次比较指针所在位置的高度,使高度较小的指针向中间移动,直到两指针重合。
正确性证明
记h[l]
和h[r]
为两指针所处位置的高度,s(l, r)
为面积
则面积计算公式为s(l, r) = min(h[l], h[r]) * (r - l)
假定h[l] < h[r]
:
- 若指针
l
向中间移动,h[l]
变大变小或不变,min(h[l], h[r])
变大变小或不变,r - l
变小,s(l, r)
可能变大 - 若指针
r
向中间移动,h[r]
变大变小或不变,min(h[l], h[r])
变小或不变,r - l
变小,s(l, r)
一定变小
因此,只有使高度较小的指针移动,才有可能找到最大值。
与此同时,此方法不会忽略可能出现的最大值,故正确。
代码
class Solution {
public:
int maxArea(vector<int>& height) {
int n = height.size();
int l = 0, r = n - 1;
int ans = 0;
while (l < r) {
ans = max(ans, min(height[l], height[r]) * (r - l));
if (height[l] < height[r])
l ++ ;
else
r -- ;
}
return ans;
}
};