给定 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
我写的是暴力版的,两层循环求任意两个柱子之间的面积最大值
超时
然后看解答,有一个双指针来求最大值
具体怎么做的呢?
初始指针i指向0位置,初始指针j指向最后一位也就是len(height) - 1位置
如果height[i] >= height[j],也就是说左边的大,右边的指针左移,即j–,如果右边的大,也就是左边的右移。直到i,j的值相同,最大值就是最终输出
代码
class Solution:
def maxArea(self, height: List[int]) -> int:
i = 0
sum_height = 0
j = len(height) - 1
while i != j:
tmp = (j - i) * min(height[i], height[j])
if sum_height < tmp:
sum_height = tmp
if height[i] >= height[j]:
j -= 1
else:
i += 1
return sum_height
怎么理解呢,这个移来移去的能保证得到的是最大值吗?
绝大部分的解释是每次移动的时候,肯定是小的向大的移动,而不是大的向小的移动,因为底大小小了1,如果是大的向小的移动,由于能装多少水只会由小的边决定,所以,不管大的向小的移动之后是大还是小都只会让最后的面积更小,但是如果小的向大的移动,如果移动之后可能会造成面积变大。
但是我觉得这个说法不是很对,这个感觉默认了这个做法是对的,然后跟你讲道理。
有一个老哥说的比较好
从最初开始:
- 指针i指向0位置,指针j指向len(height) - 1位置
- 此时如果height[i] >= height[j] (height[i] < height[j]相同),j-=1,意味着什么呢,我写成的height[i,j]形式,表示I和j之间的面积,去除了求以下面积的过程
height[i+1,j],height[i+2,j],height[i+3,j]…height[j-2,j],height[j-1,j]。
此时随着从左到右,横坐标,也就是i和j的距离慢慢变小,而有效的高度只会和height[j]相同或者更小,面积也会更小,所以height[i,j]是(height[i+1,j],height[i+2,j],height[i+3,j]…height[j-2,j],height[j-1,j]中最大的),所以排除了没必要再遍历的解。 - 所以可以得到最大的解
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/container-with-most-water
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。