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.

Note: You may not slant the container and n is at least 2.

这道题数学建模很简单，对于选择i,j，待求的面积即为$$min(a[j],a[i])*(j-i)$$

所以整个问题即转化为$$\forall i < j \in [1,2,...n]\text{, find} max(f(i, j)) = max(min(a[j],a[i])*(j-i)) $$

我们看到了这个问题的是求一个最大值，自然先得到了一个O($n^2$)的解法。

我们再看待求最大值的式子，即$$min(a[j],a[i])*(j-i)$$，乘积的两个因子之间并没有绝对的关系，所以按理说必须$i, j$全部遍历一遍才有可能找到最值。

但是这道题的特殊性在于第一个因子是取min。取min的好处就在于，对于min的操作数中比较大的那个，无论它是多少都不影响这个因子的大小。比如$a[i^*] < a[j^*]$，那么对于这个i来说（固定i），$\forall j \in [i^*+1, j^*] f(i,j) \leq f(i^*,j^*)$ （这个证明很简单），所以我们就不用计算这个$i=i^*$时其他的j值了。我们这时候将i++，再一次比较$a[j],a[i]$即可。

class Solution {
public:
    int maxArea(vector<int>& height) {
        int i = 0, j = height.size()-1, ans = 0;
        while (i < j){
            ans = max(ans, min(height[i], height[j]) * (j-i));
            if (height[i] <= height[j]){
                i++;
            } else {
                j--;
            }
        }
        return ans;
    }
};