题目描述:
给定一个长度为 n
的整数数组 height
。有 n
条垂线,第 i
条线的两个端点是 (i, 0)
和 (i, height[i])
。
找出其中的两条线,使得它们与 x
轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器
输入输出示例:
思路:对于这道题的思路很明显,我们需要找两个高度直接能容纳的容积取最大的那个,如果我们使用两层循环来实现,遍历找到每两个高度之间的容积,时间复杂度为O(n^2),不出意外的话是要超时的(也是尝试过了确实会超时),所以我们可以采用双指针left,right,计算两个指针之间的容积,然后移动指针,移动方法就是我们需要移动更小的一端,因为盛水的容积需要更小的一端来决定。以下给了两份代码,首先是使用双层循环的(超时了):
class Solution:
def maxArea(self, height: List[int]) -> int:
max_v = 0
for i in range(0,len(height)-1):
for j in range(i+1,len(height)):
if min(height[i],height[j]) * (j - i) > max_v :
max_v = min(height[i],height[j]) * (j - i)
return max_v
然后是优化之后双指针的(通过了):
class Solution:
def maxArea(self, height: List[int]) -> int:
max_v = 0
left = 0
right = len(height) - 1
while left < right :
current_v = min(height[left],height[right]) * (right - left)
max_v = max(max_v,current_v)
if height[left] < height[right] :
left += 1
else :
right -= 1
return max_v