LeetCode 11. Container With Most Water

题目：Given n non-negative integers a1, a2, …, an, where each represents a point at coordinate (i, ai). n vertical lines are drawn such that the two endpoints of line i is at (i, ai) and (i, 0). Find two lines, which together with x-axis forms a container, such that the container contains the most water.

分析：任意两个竖线（高度为$a_i$和$a_j$）所形成的容器的容量为$C_{i,j} = |i - j| * min(a_i, a_j)$。看到这道题目，最直接的想法是穷尽所有可能的 $i, j$ 组合，找出最大的容量值。显然这种$O(n^2)$的算法是不可能通过的，而所要做的就是在此基础上，找出其中不必要的计算，降低复杂度。

对于任意i，在与j(j>i)进行组合时，对于所有aj >= ai，最大值一定在距离i最远的地方，也就是说我们可以从后往前扫描j，找到第一个大于等于ai的aj，前面所有的比ai大的aj，我们都不需要进行计算。

再往前走一步，对于任意i和从后往前第一个aj >=ai的j，$C_{i, j} = (j - i) * a_i$。对于i和j之间的任意k，如果ak < ai，那么$C_{i,k} = (k - i) * a_k < C_{i, j}$，如果ak >= ai, $C_{i,k} = (k - i) * a_i < (j - i) * a_i = C_{i, j}$。如果想找到比$C_{i,j}$更大的值，唯一的可能是j后面的那些$a_n$。所以我们现在只需要对于任意的i，从后往前扫描到第一个大于ai的aj，并记录过程中最大的容量值。看起来好像可以省去大部分计算，但在最坏情况下，例如一个降序排列的a1, a2, …, an，时间复杂度依然为$O(n^2)$。

对于任意i，有没有必要从最后开始扫描呢？对于第一个大于ai的aj之后的an，an < ai，对于i和j之间的任意k，如果ak >= ai，那么显然$C_{k,n} = (n - k) * a_n < (n - i) * a_n = C_{i,n}$。
而如果ak < ai，那么$C_{k,n} = (n - k) * min(a_k, a_n) <= (n - k) * a_n < (n - i) * a_n = C_{i,n}$。也就是说想找到更大的，只需要从j处开始扫描即可。这样就有了$O(n)$的算法。

#include <vector>
#include <algorithm>
using namespace std;

class Solution {
public:
    int maxArea(vector<int>& height) {
        int size = height.size();
        int area = 0, maxArea = 0, i = 0, j = size - 1;

        for (int i = 0; i < j; i++) {
            for (; j > i; j--) {
                area = (j - i) * min(height[j], height[i]);
                maxArea = max(area, maxArea);
                if (height[j] > height[i]) { break; }
            }
        }
        return maxArea;
    }
};

后话：一开始其实没有想到最后一步，看了LeetCode Editorial Solution的解法，才想明白了要怎么优化。代码量少，但想明白有点难。