题目描述:
给定 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
'''
这道题难度适中。听说用C或C++能暴力破解,然鹅Python是不存在能够暴力的时间的。
暴力法:
找出所有的面积,比较一下,答案就可以出来。时间复杂度O(n²)
双指针法:
在0,len(s)-1处设立两个指针i,j,往中间移动,试图找出i , j且i < j,使得min(s[i],s[j]) * (j - i)取得最大的面积
问题来了,i,j什么时候才能移动呢。
观察一下发现,当s[i] < s[j]时,若j往中间移动,就算有s[j-n] > s[j],那会因为受限于s[i]的大小,使得它的面积无法变大。
当s[i] > s[j]时,若j往中间移动,此时间隔变小,而当s[j-n] > s[j]时,则有可能有(s[i],s[j-n]) * (j - n - i)>(s[i],s[j]) * (j - i)。
同理对于i也有如此规律。
所以得出结论长度较短的要向中间移动。根据这个规律,写出代码很容易。时间复杂度为O(n)
优化的双指针法:
若存在i,j且i < j,使得min(s[i],s[j]) * (j - i)取得最大的面积,则对于任意a < i,不存在s[a]>s[i].
如果(j - a) > (j - i) 并且 s[a] > s[i],则有(s[a] , s[j]) * (j - a) > (s[i],s[j]) * ( j - i)。而(s[i],s[j]) * ( j - i)已经是最大面积,因此不存在任意a < i,s[a] > s[i]。
同理对于b > j,不存在s[b] > s[j]
根据以上规律可以增加i , j的更新规律。即长度较短的要向中间移动且,比短者还要短的跳过。
虽然时间复杂度仍然为O(n),但是省去了多次求面积的过程使得算法的运行速度加快
下面是优化的双指针法
'''
class Solution:
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
i,j,m = 0,len(height)-1,0
while i < j:
h = min(height[i],height[j]) #短者
if h*(j-i) > m:
m = h*(j-i)
while height[i] <= h and i < j: i += 1 #短者还要短的跳过
while height[j] <= h and i < j: j -= 1 #短者还要短的跳过
return m