LeetCode:盛最多水的容器(Python版本)

LeetCode刷题日记

  • 盛最多水的容器
    • Python代码

盛最多水的容器

来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/

给定 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。

说明: 你不能倾斜容器,且 n 的值至少为 2。

LeetCode:盛最多水的容器(Python版本)_第1张图片

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

示例:

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

Python代码

首先看到题目的想法是直接遍历两次数组,暴力找到最大值,Python尝试后失败,可能C语言能通过吧,不晓得没尝试,贴上代码,时间复杂度为O(n2)

class Solution(object):
    def maxArea(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        Area = []
        for i in range(len(height)):
            for j in range(i, len(height)):
                Area.append(min(height[i], height[j]) * (j - i))  # 最短的板,乘距离
        return max(Area)

尝试失败后思考,如果只遍历一次能不能得到答案?这样时间复杂度就降到了O(n),尝试后可行,主要思路就是打个比方吧,两个人,一个站在最前面,一个站在最后面,谁拿到的板最短,谁向中间走一步,比较乘积的大小。

class Solution(object):
    def maxArea(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        i = 0
        j = len(height) - 1
        maxArea = 0
        while i < j:  # while退出条件,走到了中间,就不再走了
            maxArea = max(maxArea, min(height[i], height[j]) * (j - i))
            if height[i] < height[j]:  # 如果i这个人拿到了最短的板,往前走一步
                i += 1
            else:  # 否则另一个人往前走
                j -= 1
        return maxArea

执行用时 : 248 ms, 在Container With Most Water的Python提交中击败了2.53% 的用户
内存消耗 : 13.1 MB, 在Container With Most Water的Python提交中击败了0.85% 的用户

你可能感兴趣的:(Python,动态规划)