题意描述:
给定 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0)。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器,且 n 的值至少为 2。
图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
示例:
输入: [1,8,6,2,5,4,8,3,7]
输出: 49
解题思路:
Alice: 暴力搜索吗,O(n^2) 的时间复杂度会超时吧 ?
Bob: 试一试呗,不过应该有更快的解法。
Alice: 容器的值等于 长 * 高,如果在数组两侧的附近找到最高大的两个值,这时候组成的容器应该是最大的。
Bob: 那要是 较大的值 在数组的中间怎么办 ?
Alice: 所以肯定有一个搜索的过程,因为无法直接确定答案。关键是怎么压缩搜索的空间,避免重复的计算。
Bob: 难道这道题也是双指针 ? 一个指向左侧,一个指向右侧 ?
Alice: 那这两个指针的位置呢 ?如何移动呢 ?
Bob: 我想起一件事,“木桶效应”,就是说一个木桶的最大容量取决于其最短的一块木板。如果我们能向 增加 木桶的最短木板的长度的方向 移动,就能把木桶的容量增大。
Alice: 你是说 双指针的移动方式应该遵循 增加 最短木板的 长度 的原则,所以这应该是一个贪心的算法 ?
Bob: 是的,我们可以把双指针的初始位置放到 数组的两侧, 然后往中间移动。
Alice: 一直到两个指针相遇,在整个过程中的移动方式就像上面说的那样。但是这样能保证找到最优解吗 ?
Bob: 我也不知道该怎么证明呀,只是按照双指针的方式 是仅仅搜索了可以让 容器体积增加的解,应该是可以的。
Alice: 有道理,还是先试试吧。
Bob: 😎😎。
代码:
Python 方法一: 暴力求解 =》 超时。
class Solution:
def maxArea(self, height: List[int]) -> int:
ans = 0
for x in range(0, len(height)):
for z in range(x+1, len(height)):
if min(height[x], height[z]) * (z-x) > ans:
ans = min(height[x], height[z]) * (z-x)
return ans
Java 方法一: 暴力求解 =》 通过。
class Solution {
public int maxArea(int[] height) {
int tmp = 0;
int ans = 0;
for(int i=0; i<height.length; ++i){
for(int j=i+1; j<height.length; ++j){
tmp = Math.min(height[i], height[j]) * (j-i);
if(tmp > ans){
ans = tmp;
}
}
}
return ans;
}
}
Python 方法二: 双指针。
class Solution:
def maxArea(self, height: List[int]) -> int:
left = 0
right = len(height) - 1
maxArea = -1
minHeight = 0
while left < right:
minHeight = min(height[left], height[right])
if maxArea < minHeight * (right - left):
maxArea = minHeight * (right - left)
if height[left] < height[right]:
# 朝着可能让面积增大的方向移动
left += 1
else:
right -= 1
return maxArea
Java 方法二: 双指针。
class Solution {
public int maxArea(int[] height) {
int maxArea = -1;
int left = 0;
int right = height.length - 1;
while(left < right){
maxArea = Math.max(maxArea, Math.min(height[left], height[right]) * (right - left));
if (height[left] < height[right]){
left ++;
}else{
right --;
}
}
return maxArea;
}
}
易错点:
- 一些测试用例:
[1,8,6,2,5,4,8,3,7]
[1,8]
[1,0,9,9,0,1]
[0,0,9,9,0,120]
- 答案:
49
1
9
27
总结: