题目:
给你 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。
示例:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
题目的理解:
计算两个数之间的面积,获取到最大的面积。
from typing import List
class Solution:
def maxArea(self, height: List[int]) -> int:
maxValue = 0
for index, value in enumerate(height):
for i in range(index + 1, len(height)):
water = (i - index) * min(value, height[i])
maxValue = max(maxValue, water)
return maxValue
时间复杂度为O(n!), 当数量越大计算时间越长,导致计算超时。
换一个方式,从最大值进行判断两个最大值之间的容器是多大,然后计算第二大值的容器大小,一直到剩下的最大值乘以数组长度小于最大值,那么返回最大值。
from typing import List
class Solution:
def maxArea(self, height: List[int]) -> int:
water = list()
for index, value in enumerate(height):
water.append((index, value))
result = 0
while len(height) > 0:
maxValue = max(height)
height.remove(maxValue)
values = list(filter(lambda x: x[1] >= maxValue, water))
value = (values[-1][0] - values[0][0]) * min(values[-1][1], values[0][1])
result = max(result, value)
if result >= len(water) * maxValue:
break
return result
还是计算超时了,开来停止的条件还可以再优化下。
偷偷看了下提示数组,双指针,马上能够想到一个指针指向头,一个指针指向尾,那么两个指针怎么挪动呢(思考了很多种方法),挪动的条件哪个值小,哪个就向里挪动。
python实现
from typing import List
class Solution:
def maxArea(self, height: List[int]) -> int:
left = 0
right = len(height) - 1
maxValue = 0
while left < right:
maxValue = max(maxValue, min(height[left], height[right]) * (right - left))
if height[left] < height[right]:
left += 1
else:
right -= 1
return maxValue
想看最优解法移步此处
提交

写代码真的是程序员极少的一部分工作,主要是思考。
// END 突然不喜欢自己的发脾气了,应该自己受不了
网友评论