[Leet Code - Array] 11. Conatainer With Most Water
题目来源
题意分析
给定 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。画 n 条垂直线,使得垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
注意:你不能倾斜容器,n 至少是2。
解题思路
最简单的想法是对每种可能的情况进行遍历,即对所有可能的垂直线的组合,分别计算容量,然后取其中的最大值。但实际上这种算法是有问题的。该算法法主体由两层嵌套循环组成,时间复杂度为O(n^2),显然这种算法不够好。
注意到如果从取值的两端向中间遍历,若是两条垂直线中的一条短于另一条,则之后保留该条垂直线的所有容器,容量都不可能超过该容器,因为随着遍历向中间进行,容器的长度在缩小,而这样遍历得到的容器的高度不可能超过初始高度,从而体积也更小。因此之后的遍历若想得到更大的容器,则一定不能保留较短的那条垂直线。这导出了如下算法:遍历由列表的两端向中间进行,每次较短的那条垂直线向中间移动一个单位,由对应的另一条垂直线代替;每次计算对应容器的容积,保留这些容积中的最大值;当两条垂直线靠在一起时,遍历结束,此时已经得到了可能的最大容积。这种算法只需进行一层循环,时间复杂度为O(n),因而更好。
答题源码
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
result = 0
i = 0
j = len(height) - 1
while i < j:
result = max(result, min(height[i], height[j]) * (j - i))
if height[i] < height[j]:
i += 1
else:
j -= 1
return result