题目描述:
给定一个长度为 n
的整数数组 height
。有 n
条垂线,第 i
条线的两个端点是 (i, 0)
和 (i, height[i])
。
找出其中的两条线,使得它们与 x
轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7] 输出:49 解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例 2:
输入:height = [1,1] 输出:1
解题思路
首先题目要求我们计算水的面积,面积计算=长*宽(高),由于程序输入是个数组,所以长就为右边端点索引减去左边端点索引,高就为两边高较短的(因为要防止水移除)所以我们要从左到右依次改变端点的位置,根据高以及端点的索引去计算更新面积,最终返回结果。
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
left=0
right=len(height)-1
result=0
while left<right:
water = min(height[left],height[right])*(right-left)
if water>result:
result=water
if height[left]<height[right]:
left+=1
else:
right-=1
return result
-
right=len(height)-1为右边端点位置,因为len(height)为数组长度,左右端点需要索引从0开始的。
-
如果
height[left] < height[right]
,则移动左指针left += 1
。因为如果移动右指针,容器的高度不会增加,而宽度会减小,所以水容量只会减少。 -
如果
height[left] >= height[right]
,则移动右指针right -= 1
。同理,移动左指针不会增加水容量。