给定 n 个非负整数 a1,a2,...,an,每个数代表坐标中的一个点 (i, ai) 。画 n 条垂直线,使得垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
注意:你不能倾斜容器,n 至少是2。
思路一:暴力法(这种方法会超时,但是很多动态规划都是这个思路,所以记录一下),维护一个一维数组dp,其中dp[i]表示必须以下标i为右边界的矩形最大面积,那么需要两层循环,外层循环控制当前走到下标是第i个,内层循环遍历小于i的所有矩形的面积,用res记录所有dp的最大值,最后返回。
参考代码:
class Solution {
public:
int maxArea(vector<int>& height) {
vector<int> dp(height.size(), 0);
int res = 0;
for (int i = 1; i < height.size(); i++) {
for (int j = 0; j < i; j++) {
dp[i] = max(dp[i], (i - j)*(min(height[i], height[j])));
}
res = max(dp[i], res);
}
return res;
}
};
思路二:这道题还可以用O(n)的时间复杂度,O(1)的空间复杂度来完成,初始化计算出0为左边界,height.size()-1为右边界的矩形面积,然后比较左右边界哪个小,把小的边界向内收缩,如序列:
1 8 6 2 5 4 8 3 7计算过程如下图所示:
为什么可以这么做的思想在于如果一定长的一边收缩,那么得到的举行面积一定不会比当前的面积大,因为矩形面积取决于最短的那条边。当然还有一种矩阵的解释(如下图所示),行代表左边界,列代表右边界,所以我们的解空间在右上角(非x代表的区域,并且初始化为O的位置,代表左边为1,右边为6):
1 2 3 4 5 6
1 x ------- o
2 x x
3 x x x
4 x x x x
5 x x x x x
6 x x x x x x
假如左边界比右边界短,那么如果把右边界向内移一定不会得到更好的解,所以如上图所示[1,2],[1,3],[1,4],[1,5]一定不会比当前解更好(这些解已经无效,用-表示),于是向下移,如下图所示:
1 2 3 4 5 6
1 x ------- o
2 x x o
3 x x x |
4 x x x x |
5 x x x x x |
6 x x x x x x
由于这时右边的边比较短,所以如果左边的边内移也一定不会得到更优的解,所以[3,6],[4,6],[5,6]这些解被消去,在经过很多次步骤之后,找出来一条路径(O所示的路径),表示最优的解一定在这条路径上,如下图所示。
1 2 3 4 5 6
1 x ------- o
2 x x - o o o
3 x x x o | |
4 x x x x | |
5 x x x x x |
6 x x x x x x
我们观察到其他解空间已经全部搜索过了或者被消去了,即整个解空间已经搜索完了,不可能出现另外一条路径存在最优解。
参考代码:
class Solution {
public:
int maxArea(vector<int>& height) {
int left = 0, right = height.size() - 1;
int res = (right-left)*(min(height[left],height[right]));
while (left < right) {
bool leftShort = height[left] < height[right] ? true : false;
if (leftShort) {
left++;
}
else {
right--;
}
res = max(res, (right - left)*(min(height[left], height[right])));
}
return res;
}
};