前言
难怪LeetCode OJ在找工作时被很多人推荐,发现了这道最大蓄水题目就是美团的笔试最后一道题,当时我霸笔只有这一道题目没有答出来,因此也就没有获得面试机会,可惜了。
题目
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.
思路
首先,我需要模拟出这个场景,推荐一个画图软件: http://www.processon.com,虽然很小众,但是确实很好用
当从两边向中间考虑时,乘水的面积是由(两端较小的高度)×(两个板之间的宽度)决定的。
假定初始的盛水面积为ans=0,lh为左边的高度,rh为右边的高度,如果lh < rh, 则向右运动,寻找第一个比当前lh大的左节点。同理,如果lh > rh,则向左运动,寻找第一个比当前rh大的右节点。
截止条件为坐标L >= R。
AC代码
public class Solution {
public int maxArea(int[] height) {
int i, j, lh, rh, area, tmp, len = height.length;
lh = height[0];
rh = height[len - 1];
area = 0;
i = 0;
j = len - 1;
while (i < j) {
tmp = Math.min(lh, rh) * (j - i);
if (tmp > area) {
area = tmp;
}
if (lh < rh) {
while (i < j && height[i] <= lh) {
i ++;
}
if (i < j) {
lh = height[i];
}
} else {
while (i < j && height[j] <= rh) {
j --;
}
if (i < j) {
rh = height[j];
}
}
}
return area;
}
}