给你 n 个非负整数 ,每个数代表坐标中的一个点 。在坐标内画 n 条垂直线,垂直线 的两个端点分别为 和 。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器,且 n 的值至少为 2。
图中垂直线代表输入数组 。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
根据木桶原理可知,容器中水的最大容量受限于最短的边。因此,设置双指针left
和right
分别指向数组的两端,新建变量maxArea
保存当前最大的容量。
- 如果
left
指向的边较短,则计算此时的容量并更新maxArea
,left
向右移动 - 如果
right
指向的边较短,则计算此时的容量并更新maxArea
,right
向左移动
class Solution:
def maxArea(self, height: List[int]) -> int:
if height == []: return 0
left, right = 0, len(height) - 1
maxArea = 0
while left < right:
# 计算此时的面积,受限于最短边
if height[left] < height[right]:
area = height[left] * (right - left)
if area > maxArea:
maxArea = area
left += 1
else:
area = height[right] * (right - left)
if area > maxArea:
maxArea = area
right -= 1
return maxArea