给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例 2:
输入:height = [1,1]
输出:1
提示:
n == height.length
2 <= n <= 105
0 <= height[i] <= 104
蛮力法
以为优化一下能过结果还是超时了
class Solution:
def maxArea(self, height: List[int]) -> int:
res, length = [], len(height)
for i, h in enumerate(height):
_max = (length - 1 - i) * min(height[i], height[length - 1])
if h <= height[-1]:
res.append(_max)
continue
# 在 p 以及P之前没有更高的高度,就返回 _max
p = _max // h + i
j = length - 2
while j >= p:
_tmp = (j - i) * min(h, height[j])
if _tmp <= _max:
j -= 1
else:
_max = _tmp
p = _max // h + i
res.append(_max)
return max(res)
双指针
两头往中间移动,最开始底最宽,移动的时候先移动低的那边,交替移动。
class Solution:
def maxArea(self, height: List[int]) -> int:
left,right, length = 0, len(height)-1, len(height)
res = 0
while left < right:
res = max((right - left) * min(height[left], height[right]), res)
if height[left] >= height[right]:
right-=1
else:
left+=1
return res