题目
给定一个长度为 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
提示:
n == height.length
2 <= n <= 105
0 <= height[i] <= 104
解题思路
用双指针,在数组开始设置一个指针,结尾设计一个指针,并计算此时的面积,然后比较两个指针位的值,哪边小移动哪边的指针。且移动的时候如果下一个值小于之前的值,则继续移动指针,只有下一个值大于之前的值在进行计算面积并比较之前计算的面积,选最大值。不断重复上述步骤,直到左指针等于右指针。
代码
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
result = 0
left = 0
left_num = height[left]
right = len(height) - 1
right_num = height[right]
while left < right:
area = (right - left) * min(height[left],height[right])
result = max(area,result)
if height[left] <= height[right]:
while left < right:
left += 1
if height[left] < left_num:
continue
else:
left_num = height[left]
break
else:
while left < right:
right -= 1
if height[right] < right_num:
continue
else:
right_num = height[right]
break
return result