Leetcode 11. 盛最多水的容器(Python3)

11. 盛最多水的容器

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

暴力法：TLE，仅供参考思路

class Solution:
    def maxArea(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        res = 0
        for i in range(0,len(height)-1):
            for j in range(i+1,len(height)):
                s = (j-i)*min(height[i],height[j])
                res = max(s,res)
        return res

优化：双指针法

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