题目
给定 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
如下图所示。
代码
第一遍无脑使用暴力搜索,然后超时:
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
minNum = 0
for i in range(len(height)):
for j in range(len(height)):
if i >= j:
continue
if abs(i - j) * min(height[i], height[j]) > minNum:
minNum = abs(i - j) * min(height[i], height[j])
return minNum
看了题解,使用双指针,:
class Solution(object):
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
maxarea = 0
x, y = 0, len(height) - 1
while x < y:
maxarea = max((y-x) * min(height[x], height[y]), maxarea)
if height[x]<height[y]:
x+=1
else:
y-=1
return maxarea