给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
说明:你不能倾斜容器。
最开始采用的是暴力法,结果不出所料,超时了
class Solution:
def maxArea(self, height: List[int]) -> int:
max = 0
i = 0
while i < len(height) - 1:
j = i + 1
while j < len(height):
if (j - i)*min(height[i],height[j]) > max:
max = (j - i)*min(height[i],height[j])
j+=1
i+=1
return max
换双指针,蛮简单的
具体思路是,两个指针一个从左往右,一个从右往左,先判断面积,然后比较两个指针对应的值谁更小,更小的指针移动,另一个不动,直到俩指针相遇
class Solution:
def maxArea(self, height: List[int]) -> int:
max = 0
i = 0
j = len(height)-1
while i < j:
if (j - i)*min(height[i],height[j]) > max:
max = (j - i)*min(height[i],height[j])
if height[i]<=height[j]:
i+=1
else:
j-=1
return max
别人的代码,同一思路
class Solution:
def maxArea(self, height: List[int]) -> int:
i, j, res = 0, len(height) - 1, 0
while i < j:
if height[i] < height[j]:
res = max(res, height[i] * (j - i))
i += 1
else:
res = max(res, height[j] * (j - i))
j -= 1
return res