给定 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器,且 n 的值至少为 2。
输入: [1,8,6,2,5,4,8,3,7]
输出: 49
方法一:
暴力法:
复杂度过高
array=list(int(i) for i in input().split())
def maxArea(array):
length=len(array)
max_capacity=0
for start in range(length-1):
for end in range(start+1,length):
if array[start]>array[end]:
height=array[end]
else:
height=array[start]
capacity=(end-start)*height
if capacity>max_capacity:
max_capacity=capacity
return max_capacity
print(maxArea(array))
时间复杂度:O(n^2),
空间复杂度:O(1)
方法二:
双坐标,每次向高一点的坐标逼近
array=list(int(i) for i in input().split())
def maxArea(height):
length=len(height)
max_area=0
start=0
end=length-1
while start<end:
max_area=max(max_area,min(height[start],height[end])*(end-start))
if height[start]>height[end]:
end-=1
else:
start+=1
return max_area
print(maxArea(array))
时间复杂度:O(n),一次扫描。
空间复杂度:O(1),使用恒定的空间。