萌新练习写代码的每日一练:盛最多水的容器
给你 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题