力扣:11. 盛最多水的容器(Python3)

题目:

给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。

找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。

返回容器可以储存的最大水量。

说明:你不能倾斜容器。

来源:力扣(LeetCode)
链接:力扣

示例:

示例 1:

输入:[1,8,6,2,5,4,8,3,7]

力扣:11. 盛最多水的容器(Python3)_第1张图片


输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。


示例 2:

输入:height = [1,1]
输出:1

解法:

设置双指针,一个位于首部(head), 一个位于尾部(tail)。head,tail谁小谁就向中间移动,因为一开始容器的长(tail-head)已经最大,限制因素是容器的宽(min(height[head], height[tail]),height[head], height[tail]简写为t4,t5),所以此时需要扩大容器的宽,也就是提高t4,t5中小的值,所以小的一方向中间移动,寻找下一个比当前值大的元素,循环终止条件是head和tail挨在一起,也就是head=tail-1。在循环过程中,如果在中间区域找不到更大的元素,那么容器的宽无法提高,那么就可以提前结束循环。

知识点:

1.enumerate():传入的参数是可迭代对象,返回的值是小标和元素,比如(0, seq[0]), (1, seq[1]), (2, seq[2]),可用于for循环。

2.关于列表切片最后一个参数:切片中最后一个元素默认值1,表示步长。当为-1时,表示逆序,比如list = [3, 4, 6, 7, 2, 10, 16],list[5:0:-2]输出为10, 7, 4,5表示从下标5开始(闭区间),0表示到下标0结束(开区间)。也可以写成list[0:5][::-2]。

3.max():传入的参数可以是多个值,也可以是可迭代对象,返回最大值,min()同理。

代码:

class Solution:
     def maxArea(self, height: List[int]) -> int:
        head = 0
        tail = len(height) - 1
        s = min(t4 := height[head], t5 := height[tail]) * (tail - head)
        while head < tail - 1:
            if t4 <= t5:
                if t4 >= max((t := height[head + 1:tail])):
                    break
                else:
                    for index, num in enumerate(t):
                        if num > t4:
                            head += index + 1
                            t4 = num
                            break
                    s = t2 if (t2 := min(t4, t5) * (tail - head)) > s else s
            else:
                if t5 >= max((t3 := height[head + 1:tail][::-1])):
                    break
                else:
                    for index, num in enumerate(t3):
                        if num > t5:
                            tail -= index + 1
                            t5 = num
                            break
                    s = t2 if (t2 := min(t4, t5) * (tail - head)) > s else s
        return s

你可能感兴趣的:(LeetCode,leetcode,算法,python)