题目
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.
思路一
遍历全部。时间复杂度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不需要往右移。