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.
如果用暴力两个for循环做,肯定是超时的,我就想到了这算是一种贪心策略。
两个下标变量,分别表示数组的头部和尾部,逐步向中心推移。这个推移的过程是这样的:
假设现在是初始状态,下标变量i=0表示头部,下标变量j=height.size(),表示尾部,那么显然此时的容器的装水量取决一个矩形的大小,这个矩形的长度为j-i,高度为height[i]与height[j]的最小值(假设height[i]小于height[j])。接下来考虑是把头部下标i向右移动还是把尾部下标j向左移动呢?如果移动尾部变量j,那么就算height[j]变高了,装水量依然没有变得更大,因为短板在头部变量i。所以应该移动头部变量i。也就是说,每次移动头部变量i和尾部变量j中的一个,哪个变量的高度小,就把这个变量向中心移动。计算此时的装水量并且和最大装水量的当前值做比较。
public class Solution {
public int maxArea(int[] height) {
int area_max = 0;
int area_tmp = 0;
int i = 0, j = height.length - 1;
while (i < j) {
area_tmp = (j - i)
* (height[i] > height[j] ? height[j] : height[i]);
if (area_tmp > area_max)
area_max = area_tmp;
if (height[i] > height[j])
j--;
else
i++;
}
return area_max;
}
}