题目:给定n个非负的整数a1, a2, ..., an,(i, ai) and (i, 0)分别代表坐标(i, ai)。连接(i, ai) and (i, 0)画直线,共有n条。找出两条直线,使得两条直线与x轴形成的容器能够盛最多的水。
例: Input: [1,8,6,2,5,4,8,3,7] Output: 49
思路:这道题最简单的方法就是双层暴力循环法,使用两个for循环依次取出左右两个垂直线i,j,计算容器面积S=min(a[i],a[j])*(j-i),最后输出最大的面积,时间复杂度是O(n^2)。那么有没有更简单的方法呢?
我们知道容器面积取决于两个因素:左右两条垂直线的较小值和左右两条垂直线之间的距离。使较小值尽量大,距离尽量大,面积就会变大。因此,可以从这两方面入手来优化左右两条垂直线的选取方法。
(1) 左右两条垂直线的较小值入手:先考虑较小值尽量大,再考虑距离尽量大。
先将所有垂直线按高度排序,先取第二高的垂直线高度作为左右垂线较小值,计算面积,再选取第三大的垂直线作为较短的垂线,找到距离它距离最远且高度比她高的垂线作为另一条垂线,计算面积,据此进行下去,找到最大面积。但是这种方法需要先排序,且需摘到排序之前的位置,实现起来较为复杂。
(2) 从左右两条垂直线之间的距离入手:先考虑距离尽量大,再考虑较小值尽量大;
使用双指针,首先分别指向开始位置和最后位置,再根据左右垂直线的较小值,优化选取。这种方法实现起来比较容易。
class Solution:
def maxArea(self, height: List[int]) -> int:
len_h=len(height)
i,j,maxS,count=0,len_h-1,0,0
while(i<j):
S=min(height[i],height[j])*(j-i)
if maxS<S:
maxS=S
if height[i]<height[j]:
i+=1
else:
j-=1
return maxS