题目描述:
给你 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
示例:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/container-with-most-water
分析:
双指针问题,我们可以设置两个指针,一个指向容器的左垂线,一个指向容器的右垂线。容器盛多少水是由两垂线的距离和较短垂线的高度决定的,设置一个变量result表示最终结果,计算出当前指针组成的容器的容量,如果容量大于result则更新result的值,并且移动指向较短垂线的指针
代码:
class Solution:
def maxArea(self, height):
length = len(height)
left, right = 0, length-1 # left和right分别指向容器的左右垂线
result = (right - left) * min(height[left], height[right])
while left < right:
if height[left] < height[right]: # 左垂线较短则移动左指针
left += 1
s = (right - left) * min(height[left], height[right])
if result < s:
result = s
else: # 右垂线较短,移动右指针
right -= 1
s = (right - left) * min(height[left], height[right])
if result < s:
result = s
return result