原题链接:https://leetcode.com/problems/container-with-most-water
最开始的想法是O(
n2
n
2
)的,即暴力的双层循环,比较哪两条线围成的梯形或矩形能装最多的雨水(容量视短线而定),但提交后超时,就寻找更为优化的方法。
之后的想法是先锁定长方形的最大宽度len(height)-1,计算maxArea,然后再更改low或high中边较短的那一条边,原因是在最大宽度缩短的情况下,更改较短的边才有可能会使maxArea变大,时间复杂度O(n)。
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
maxA = 0
low = 0
high = len(height)-1
while(low<high):
maxA = max(maxA,(high-low)*min(height[low],height[high]))
if(height[low] < height[high]):#谁短不要谁
low +=1
else:
high-=1
return maxA