算法设计与分析(8)
题目: 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.
算法思路:
这个问题的本质是,有n个点(k,ak),1≤k≤n。我们找出其中两个点,(i,ai),(j,aj),1≤i≤j≤n,使得 (j - i)*Min(ai, aj) 最大。这里可能很容易会想到使用动态规划的办法,但这种方法复杂度高,我们这里直接给出一个O(n)的简单求解算法。
1.n个点是存放在数组vector<int> height
中。首先,初始化 i,j 分别指向数组的头和尾。使用max记录最大值。
int i = 0, j = height.size() - 1;
int max = 0;
2.然后使用while(i < j)
循环,并用 ht = Min(height[i], height[j])。
(1)然后如果 (j - i) * ht > max成立, 那么更新max:
(2)如果 height[i] <= ht
或者 height[j] <= ht
成立,那么为了能够找到更大的max值,我们分别使用 i++和 j–。当条件(i ≥ j)成立时,我们已经遍历完成,同时找出了max的最大值。
while (i < j)
{
int ht = Min(height[i], height[j]);
if ((j - i) * ht > max)
max = (j - i) * ht;
while (height[j] <= ht && i < j) --j;
while (height[i] <= ht && i < j) ++i;
}
实现代码:
int Min(int n, int m)
{
return n <= m ? n : m;
}
int maxArea(vector<int>& height) {
int i = 0, j = height.size() - 1;
int max = 0;
while (i < j)
{
int ht = Min(height[i], height[j]);
if ((j - i) * ht > max)
max = (j - i) * ht;
while (height[j] <= ht && i < j) --j;
while (height[i] <= ht && i < j) ++i;
}
return max;
}
int main()
{
int a[7] = { 7,6,5,4,3,2,1 };
vector<int> height(a, a + 7);
int result = maxArea(height);
cout << result << endl;
return 0;
}