给定 n 个非负整数 a1,a2,...,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器,且 n 的值至少为 2。
最直观、粗暴的方法:全遍历,时间复杂度为O(n^2)。
def maxArea(height):
"""
:type height: List[int]
:rtype: int
"""
max = -1
for x1 in range(len(height)):
for x2 in range(x1 + 1, len(height)):
area = (x2 - x1) * min(height[x1], height[x2])
if area > max:
max = area
return max
方法二:有点类似于动态规划的方法。
首先,两条线条之间的容积为(x2 - x1) * min(height[x1], height[x2])
先取最左边直线和最右边直线,这种情况用(0, n-1)表示。下一步,考虑减少一个单位的宽度的情况,这时候有两种选择:(0, n-2)和(1, n-1)。对比这两种情况,如果线条的高度height[0] < height[n-1],也就说是第0根线条是“短板”,那么(0, n-2)的容积肯定会比(0, n-1)的容积小(前者容积为height[0]乘以宽度,后者的“短板”肯定小于或等于height[0],而宽度变小),因此(0, n-2)的情况可以不用比较,因此,下一步考虑(1, n-1)的情况,依次类推。时间复杂度为O(n)。
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
left = 0
right = len(height) - 1
max_area = -1
while left < right:
tmp_area = (right - left) * min(height[left], height[right])
if tmp_area > max_area:
max_area = tmp_area
if height[left] > height[right]:
right -= 1
else:
left += 1
return max_area