给你 n 个非负整数 a 1 , a 2 , . . . , a n a_1,a_2,...,a_n a1,a2,...,an,每个数代表坐标中的一个点 ( i , a i ) (i, a_i) (i,ai)。在坐标内画 n 条垂直线,垂直线 i i i的两个端点分别为 ( i , a i ) (i, a_i) (i,ai) 和 ( i , 0 ) (i, 0) (i,0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器,且 n 的值至少为 2。
图中垂直线代表输入数组 [ 1 , 8 , 6 , 2 , 5 , 4 , 8 , 3 , 7 ] [1,8,6,2,5,4,8,3,7] [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 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