【LeetCode】11.盛最多水的容器

问题描述

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

说明：你不能倾斜容器。

示例 1：

输入：[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

提示：

• n == height.length
• 2 <= n <= 105
• 0 <= height[i] <= 104

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/container-with-most-water

法I：暴力求解（超时了……）

//法I：暴力求解
class Solution {
public:
	int maxArea(vector<int>& height) {
		int max = 0;
		for (int i = 0; i < height.size(); i++) {
			for (int j = i + 1; j < height.size(); j++) {
				int h = height[i] > height[j] ? height[j] : height[i];
				int tmp = (j - i)*h;
				max = tmp > max ? tmp : max;
			}
		}
		return max;
	}
};

法II：双指针【重点】

这是面试中十分经典的一个问题。我们首先给出解题思路，然后给出正确性的证明。

（1）解题思路

我们以下面为例进行分析：

[1,8,6,2,5,4,8,3,7]

我们用两个指针i和j分别指向这个数组的首和尾，用height[i]和height[j]表示其高度。因此我们给出面积的计算公式：s=min(height[i],height[j])*(j-i)。可以看出，面积的大小取决于两个指针之间的距离和高度中的短板。

接下来我们移动其中一个指针（我们假设指针总是向内移动），那么我们应该移动哪一个呢？首先我们要明确一点是，无论移动哪个指针，因为都是向内移动，因此两指针之间距离一定减小了。我们假设height[i]<height[j]，如果我们移动j指针（向内移动），分两种情况考虑：

1. 若height[j-1]<height[j]，则此时的短板是min(height[i],height[j-1])，相比于之前不变或减小，因此s一定减小；
2. 若height[j-1]>height[j]，则此时的短板是仍然是height[i]，相比于之前不变，因此s也一定减小；

所以我们知道，如果移动j指针（高度较高的那个指针），面积一定会减小，那么我们尝试移动i指针呢？

1. 若height[i+1]>height[i]，则此时短板为min(height[i+1],height[j])，相比于之前增大了，由于指针距离减小，因此面积变化不一定；
2. 若height[i+1]<height[i]，则此时短板为height[i+1]，相比于之前减小了，因此面积也会减小。

由上面分析我们可知，移动i指针，面积的变化不定，但是有增大的可能，因此我们选择永远移动高度较小的那个指针。（有点像博弈论了）。

（2）正确性证明

那么为什么双指针一定能找到最优解呢？下面我们给出正确性的证明。

我们知道穷举法是把所有可能的情况一一算出，之后比较取得最大面积的，假若数组中有n个元素，那么共有C(n,2)种可能情况。

从指针指向数组的头和尾的第一步来考虑，看看排除了哪些情况：
指针i指向数组头，指针j指向数组尾，之后指针i向内移动（我们假设height[i]<height[j]），我们认为重新定义了数组的边界为i+1和j，相当于把索引i指向的元素剔除出数组。则此次移动排除了指针i作为左边界，从i+1到j作为右边界的所有情况。经过上面的分析我们就可以知道，剔除的这些情况对应的面积都没有s[i:j]大，因此剔除这些情况对于求最大面积是没有影响的。每次移动完指针之后，我们都理解为得到了一个全新的数组，而我们的两个指针刚好指向了这个新数组的边界。则正确性可证。

（3）源码及结果

//法II：双指针
class Solution {
public:
	int maxArea(vector<int>& height) {
		int i = 0;
		int j = height.size() - 1;
		int s = 0;
		while (i < j) {
			int tmp = min(height[i], height[j]) * (j - i);
			s = s > tmp ? s : tmp;
			if (height[i] < height[j]) {
				i++;
			}
			else {
				j--;
			}
		}
		return s;
	}
};