萌新练习写代码的每日一练:盛最多水的容器

给你 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

思路:用双指针来做,从图中例子来看,一开始指针指向的是1和7,目前可以确定的是,选择这个宽度求出的面积一定比后续要大,面积为1*8=8,设定一个变量area_max来记录。接下来移动柱子,这个是很核心的一个思路,假设移动左边的柱子,是可以增大面积的,反过来如果移动右边的柱子,效果甚至不如不移动。所以得出移动柱子只和当前柱子高度有关系。因此可以设立两个指针i j,

if height[i] < height[j]:
    i += 1
else:
    j -= 1

直接解决了移动哪一根柱子的问题,之后用while i<j来判断柱子是否相遇,在相遇之前一直记录最大面积,然后与之前相比较,完成!

代码:

class Solution:
    def maxArea(self, height: List[int]) -> int:
        res = 0
        n = len(height)
        i = 0
        j = n-1
        while i<j:
            area_max = min(height[i], height[j]) * (j - i)
            if area_max > res:
                res = area_max
            if height[i] < height[j]:
                i += 1
            else:
                j -= 1
        return res

每次移动i或j的时候,就说明我们已经处理好了当前柱子的所有情况,无论他是被计算的还是被放弃的。

题目来自LeetCode第11题