Container With Most Water

题目

Given n non-negative integers a₁, a₂, ..., a_n, where each represents a point at coordinate (i, a_i). n vertical lines are drawn such that the two endpoints of line i is at (i, a_i) 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.

思路一

遍历全部。时间复杂度O(N2)

class Solution {
public:
    int maxArea(vector<int> &height) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int num = height.size();
        int maxarea = 0;
        for(int i=1;i<num;i++){
            for(int j=0;j<i;j++){
                int area = min(height[j],height[i])*(i-j);
                if(area>maxarea)
                    maxarea = area;                
            }
        }
        return maxarea;        
    }
};

小数据可以过，大数据时 Run Status: Time Limit Exceeded

对上述代码剪枝优化一下：

因为在当前最大值是 maxarea 的情况下，

（1）若 height[j]>height[i] ，则 j 的右边界是 region = i-maxarea/height[i] ；

（2）若 height[j]<height[i] ，则 j 的右边界比 region 还要小。

所以 j 的右边界是 region = i-maxarea/height[i] ；

class Solution {
public:
    int maxArea(vector<int> &height) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int num = height.size();
        int maxarea = 0;
        for(int i=1;i<num;i++){
            if(height[i]==0) continue;
            int region = i-maxarea/height[i];
            for(int j=0;j<region;j++){
                int area = min(height[j],height[i])*(i-j);
                if(area>maxarea)
                    maxarea = area;                
            }
        }
        return maxarea;        
    }
};

此时可以通过大数据，但是时间复杂度还是 O(N2) 的，所以依旧耗时：Run Status: Accepted!   Program Runtime: 1172 milli secs

 

思路二

时间复杂度O(N)

class Solution {
public:
    int maxArea(vector<int> &height) {
        // Start typing your C/C++ solution below
        // DO NOT write int main() function
        int num = height.size();
        int maxarea = 0;
        int left = 0, right = num-1;
        while(left<right) {
            int area = min(height[left],height[right])*(right-left);
            if(area>maxarea)
                maxarea = area;   
            if(height[left]<height[right])
                left++;
            else
                right--;            
        }
        return maxarea;        
    }
};

上述算法的证明：

两边设一个指针，然后计算area，如果height[i] <= height[j]，那么i++；

因为在这里height[i] 是瓶颈，j 往里移只会减少面积，不会再增加area。

当height[i] <= height[j]时，为什么是i++，而不是j++来获取可能更多的水。

这是一个贪心的策略，每次取两边围栏最矮的一个推进，希望获取更多的水。

不严格的递推证明：

当height[i] <= height[j]时，为什么是i++，而不是j++来获取可能更多的水？

假设j' > j，之所以j'往左移，是因为存在height[i'] > height[j'] (i’ <= i), 而那时area' = (j' - i') * min(height[i'], height[j'])，

因为height[j'] == min(height[i'], height[j'])，所以area' = (j' - i') * height[j']。

而i 和 j'构成的面积area = (j' - i) * min(height[i], height[j'])。

area' >= area，所以j不需要往右移。