Problem: 11. 盛最多水的容器
题目
给定一个长度为 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
思路
我们很容易想到计算两个柱子的水容量的公式:
a
n
s
=
m
i
n
(
l
,
r
)
∗
(
r
−
l
)
ans = min(l,r)*(r-l)
ans=min(l,r)∗(r−l)
一种朴素的解法是两层循环,计算每两根柱子之间的容量,保留最大值。
这种方法肯定是可行的,但是我们思考下有没有优化的方案。
对于水容量而言, r − l r-l r−l 肯定是越大越好,高度也是越高越好,我们的核心思想是:找容量尽可能高的组合,掠过一些可以跳过的;
我们不妨一开始从左右边界开始遍历,此时 r − l r-l r−l一定是最大的,我们计算左右边界的最小值,并且算出最终容量,现在到了移动边界的时候。
如果我们此刻移动更高的边界,那么如果下一个边界更高,水容量也不会更高,因为此时 r − l r-l r−l变小了,而高度依旧是那个低的边界的高度,这种组合不应该被遍历
所以我们要移动低的那个柱子,因为下一个容量更高的组合只可能在低的柱子移动后出现。循环终止的边界是 l = = r l==r l==r
我们遍历了所有可能情况,保留最大值即可。
复杂度
时间复杂度:
O ( n ) O(n) O(n)
空间复杂度:
O ( 1 ) O(1) O(1)
Code
class Solution:
def maxArea(self, height: List[int]) -> int:
l, r = 0, len(height) - 1
ans = 0
while l < r:
area = min(height[l], height[r]) * (r - l)
ans = max(ans, area)
if height[l] <= height[r]:
l += 1
else:
r -= 1
return ans