Leetcode 11.盛水最多的容器题解

Leetcode 11.盛水最多的容器题解

题目链接：

https://leetcode-cn.com/problems/container-with-most-water/

题目描述

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

思路

一个容器能盛多少水是由其两边的高度决定的。所以我最开始的思路是先固定容器的其中一边，然后去寻找可以得到最大容量的另一边。但这样做不管怎么剪枝，时间复杂度都是o(n²)，跟遍历的方法没有任何区别（这就很气 (╯`□′)╯(┴—┴）。于是我只能换个思路去想这个问题，那么问题来了，怎么换思路呢？让我们细品一下，我之前的思路其实是先确定容器的两边再往容器里注水求得容积，那既然要换一种思路，不妨把把这两步颠倒一下，先往容器里注满水再确定容器的的两边。如果我们把样例的容器注满水，可以得到下面的样子：

那么乍看之下，是不是只有三种候选容器可以成为容积最大的容器，如下图所示（有点丑，不过不要在意这些细节）。

只要再比较一下这三个容器的容积就可以求得最大容积。那么我们是怎么快速的找到这三个容器的呢，假设我们同时在每个格子间以相同的速度注水，最先满的肯定是黄色的这个容器，然后是红色最后是绿色。模拟一下这个过程，不就是一个容器满了后，比较容器两边的高度，然后把短的变往里移动找到一个比它高的边再继续灌水嘛。所以只要设置个头尾指针，这题o(n)的时间复杂度就可以解。

代码

class Solution(object):
    def maxArea(self, height):
        """
        :type height: List[int]
        :rtype: int
        """
        #头尾指针
        start, end = 0, len(height)-1
        max_count = 0
        while end > start:
            max_count = max(max_count, (end-start)*min(height[start], height[end]))
            #判断大小往里移动
            if height[end] > height[start]:
                start += 1
            else:
                end -= 1
        return max_count