双指针法解决盛最多水容器的问题

最新推荐文章于 2022-04-15 01:11:44 发布

ZeroHomer

最新推荐文章于 2022-04-15 01:11:44 发布

阅读量245

点赞数

分类专栏： Leetcode 文章标签：双指针法最大容量盛水容器算法 Python

本文链接：https://blog.csdn.net/ZeroHomer/article/details/118927987

版权

Leetcode 专栏收录该内容

1 篇文章 0 订阅

订阅专栏

双指针法

双指针法是指在遍历对象时，使用两个方向相同或者相反的指针进行扫描，从而解决相关问题。需要注意的是，这个双指针的“指针”不是指狭义的C++或C的指针，而是指广义的索引、游标、指针、迭代对象等。

盛最多水容器的问题

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

解决

使用双指针方法可以简单地解决该问题。

设置指针i，j分别指向水容器的两壁，i为左壁，j为右壁，初始化i=0，j=len（height）-1。

由题意可得，因为水容器的高度由短板决定，所以可以得到公式

$S(i,j)=min(height[i],height[j])×(j−i)$

确定公式后需要确定如何移动水容器的两板，有以下两种情况：

移动长板

若移动相对长的一边，min(height[i],height[j])不发生改变，而j-i会变小，S(i,j)会一定变小

移动短板

若移动相对短的一边，移动到的位置所在的板可能比原来的长，则min(height[i],height[j])变大，S（i，j）可能变大。

所以要选择移动短板找到最大值

实现代码如下

class Solution:
    def maxArea(self, height: List[int]) -> int:
        i,j=0,len(height)-1
        ans=0 #最终结果
        while i<j:
            if height[i]<height[j]: #左边的板较短，将左边的板向右移动
                ans = max(height[i]*(j-i),ans)
                i+=1
            else : #右边的板较短，将右边的板向左移动
                ans = max(height[j]*(j-i),ans)
                j-=1
            
        return ans