题目描述
给你 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
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/container-with-most-water
解题思路
思路一:双指针
容纳的水量是由:两个指针指向的数字中较小值 * 指针之间的距离决定。
先从题目示例开始解释双指针算法的操作过程:
题目输入的数组为:
[1, 8, 6, 2, 5, 4, 8, 3, 7]
首先左右指针分别指向数组左右两端,此时可以容纳的水量为:。
然后需要移动左右指针中对应的数较小的那个指针。
移动后变成了:
[1, 8, 6, 2, 5, 4, 8, 3, 7]
此时可以容纳的水量为。
再移动数字小的右指针:
[1, 8, 6, 2, 5, 4, 8, 3, 7]
此时可以容纳的水量为。
再移动数字小的右指针:
[1, 8, 6, 2, 5, 4, 8, 3, 7]
此时可以容纳的水量为。
两指针对应数字相同,选择其中一个。比如移动左指针:
[1, 8, 6, 2, 5, 4, 8, 3, 7]
此时可以容纳的水量为。
继续移动左指针,发现一直比右指针小所以一直移动到两指针重合。过程中对应的水量为:,,。
所以,最多可容纳的水量为49。
正确性证明:
双指针代表可以作为容器边界的所有位置的范围。一开始双指针指向数组的左右边界表示数组中所有位置都可以作为容器的边界。然后每次将对应数字较小的指针往另一个指针的方向移动,就表示这个指针不可能当容器的边界了。最后的答案就是每次以双指针为左右边界计算出的容量中的最大值。(详细证明参见官方题解)
- 复杂度分析
- 时间复杂度:,双指针总计最多遍历整个数组一遍。
- 空间复杂度:,只需要额外的常数级别的空间。
代码实现
双指针:
class Solution(object):
def maxArea(self, height):
start, end = 0, len(height)-1
maxwater = 0
while start < end:
maxwater = max(maxwater,(end - start) * min(height[start], height[end]))
if height[start] <= height[end]:
start = start + 1
else:
end = end - 1
return maxwater
Tips
双指针
出现的又一次,多练才能出直觉。