LeetCode算法题11：盛最多水的容器（Container With Most Water）

技术交流可以加：
本人微信：xcg852390212
本人qq：852390212
学习交流qq群1（已满）： 962535112
学习交流qq群2： 780902027

盛最多水的容器

LeetCode中文

LeetCode英文

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

解答

双指针法，定义两个指针l和r分别指向数组两端，两个指针向中间移动，移动规则如下：

1. 如果height[l] < height[r]，则++l
2. 否则，–r

每走一步，计算容器装水量为左右边缘较小的那个高度乘边缘的的距离min(height[l],height[r]) * (r - l),然后和结果比较去较大值。

• 时间复杂度：O(n)
• 空间复杂度：O(1)

C++代码

class Solution {
public:
    int maxArea(vector<int>& height) {
        int len = height.size();
        if(len == 0) return 0;
        
        int res = 0;
        int l = 0;
        int r = len - 1;
        while(l < r)
        {
            int tmp = min(height[l],height[r]);
            int sum = (r - l)*tmp;
            if(height[l] < height[r])
                ++l;
            else
                --r;
            
            res = max(res,sum);
            
            while(l < r && tmp == height[l]) ++l;
            while(l < r && tmp == height[r]) --r;
        }
        
        return res;
    }
};

Python代码

class Solution:
    def maxArea(self, height: List[int]) -> int:
        n = len(height)
        l = 0
        r = n - 1
        res = 0
        while l < r:
            tmp = min(height[l],height[r])
            res = max(res,(r - l) * tmp)
            if height[l] < height[r]:
                l += 1
            else:
                r -= 1
            
            while l < r and tmp == height[l]:
                l += 1
            while l < r and tmp == height[r]:
                r -= 1
            
        return res