Python求解盛水最多的容器

题目描述

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

说明：你不能倾斜容器。

LeetCode原题地址：https://leetcode-cn.com/problems/container-with-most-water/

测试用例

• 示例 1
  

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

• 示例 2

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

• 示例 3

输入：height = [4,3,2,1,4]
输出：16

• 示例 4

输入：height = [1,2,1]
输出：2

代码实现

from typing import List

class Solution(object):
    def maxArea(self, height: List[int]) -> int:
        #定义容器左右边界两个指针
        left = 0
        right = len(height) - 1
        #用来记录容器最大的面积
        max_area = 0
        while True:
            if left == right:
                break
            #计算当前容器面积的大小
            cur_area = min(height[left],height[right]) * (right - left)
            #更新指针,只更新数值小的一端
            if height[left] > height[right]:
                # 左端的高大于右端的高时就更新右指针
                right -= 1
            else:
                # 右端的高大于左端时就更新左端
                left += 1
            #更新容器的最大面积
            max_area = max(cur_area,max_area)
        return max_area

solution = Solution()
print(solution.maxArea([1,8,6,2,5,4,8,3,7]))

参考：

1. LeetCode详细题解