53 lines
1.9 KiB
C++
Raw Normal View History

2014-10-20 11:23:39 +08:00
// Source : https://oj.leetcode.com/problems/container-with-most-water/
// Author : Hao Chen
// Date : 2014-06-22
/**********************************************************************************
*
2014-10-21 11:39:53 +08:00
* 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.
*
*
**********************************************************************************/
2014-10-20 11:23:39 +08:00
class Solution {
public:
int maxArea(vector<int> &height) {
int maxArea = 0;
// two pointers scan from two sides to middle
int left = 0;
int right = height.size()-1;
int area;
while ( left < right ){
2014-12-30 22:33:05 +08:00
// calculate the area
area = (right - left) * ( height[left] < height[right] ? height[left] : height[right]);
2014-12-30 22:33:05 +08:00
// tracking the maxium area
maxArea = area > maxArea ? area : maxArea;
// because the area is decided by the shorter edge
// so we increase the area is to increase the shorter edge
2014-12-30 22:33:05 +08:00
//
// height[left] < height[right] ? left++ : right-- ;
//
// However, the above code could cause the unnecessary `area` cacluation
// We can do some improvement as below:
if (height[left] < height[right]) {
do {
left++;
} while (left < right && height[left-1] >= height[left]);
} else {
do {
right--;
} while (right > left && height[right+1] >= height[right]);
}
2014-10-20 11:23:39 +08:00
}
return maxArea;
2014-10-20 11:23:39 +08:00
}
};