(python刷题)leetcode 第11题:盛最多水的容器

题目在leetcode上的链接为:
https://leetcode-cn.com/problems/container-with-most-water/

题目描述
(python刷题)leetcode 第11题:盛最多水的容器_第1张图片
解题思路
首先很容易想到暴力法,只需要使用两重循环遍历起始位置元素 i,j,然后每遍历到以个位置时计算此时的盛水量 (j-i)*min(a[i],a[j]),并更新最大的盛水量即可。暴力法的时间复杂度为 o(n2)。
那么有不有时间复杂度更低的方法呢?
我们注意到盛水量与 两个线段起始位置的距离和最短的线段有关,我们可以使用双指针的方法来解题。首先使用两个指针 i,j 分别指向列表的开头与末尾,此时 i,j 之间的距离是最大的,为了使得盛水量尽可能增大,那么我们只需要将 a[i],a[j] 中较短的线段往较长的线段方向移动即可
为什么不是较长的线段往较短线段方向移动呢?
我们以题目为例进行说明,一开始我们的 i=0,j=8,a[i] 盛水量=(j-i)*min(a[i],a[j]),此时 j-i 减小了,而 min(a[i],a[j]) 也是小于或者等于移动前的两线段中的最小值,那么移动后的盛水量肯定会减小,因此这种方案是不对的。

ps:如果遇到 a[i] 与 a[j] 相等,即两个线段长度相同的情况,此时 i++ 或者 j-- 都可以。

复杂度分析:
由于只需要使用双指针遍历一次列表,时间复杂度为 o(n)
空间复杂度为 o(1)

python代码:

class Solution(object):
    def maxArea(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        if len(height) < 2:
            return 0
        i = 0
        j = len(height) - 1
        maxarea = 0
        while i < j:
            area = (j - i) * min(height[i], height[j])
            if area > maxarea:
                maxarea = area
            if height[i] < height[j]:
                i = i + 1
            else:
                j = j - 1
        return maxarea

你可能感兴趣的:(刷题)