编程题目:
给定 n 个非负整数 a1,a2,…,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0) 。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
输出示例:
- 输入:[1,8,6,2,5,4,8,3,7]
输出:49 - 输入:height = [1,1]
输出:1 - 输入:height = [4,3,2,1,4]
输出:16 - 输入:height = [1,2,1]
输出:2
解法:
1. 暴力搜索
时间复杂度O(N^2)
空间复杂度O(1)
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int maxArea(vector<int>& height) {
int maxVolume = 0; //记录水的最大容量
int size = height.size();
// 枚举所有间距大于等于 2 的垂直线
for(int i = 0; i < size - 1; i++)
{
for(int j = i + 1; j < size; j++)
{
// int minHeight = min(height[i], height[j])
int minHeight = height[j] > height[i] ? height[i] : height[j];
int currVolume = (j - i) * minHeight;
// int maxVolume = max(currVolume, maxVolume)
if(currVolume > maxVolume)
maxVolume = currVolume;
}
}
return maxVolume;
}
};
2. 双指针法
时间复杂度O(N)
空间复杂度O(1)
#include <iostream>
#include <vector>
using namespace std;
class Solution {
public:
int maxArea(vector<int>& height) {
int maxVolume = 0;
int left = 0; // 左指针 设置初始值为首部位置
int right = height.size() - 1; // 右指针 设置初始值为尾部位置
while(left < right)// 循环退出时left == right
{
int minHeight = height[left] > height[right] ? height[right] : height[left]; // int minHeight = min(height[l], height[r])
int currVolume = minHeight * (right - left);
// int maxVolume = max(currVolume, maxVolume)
if(currVolume > maxVolume)
{
maxVolume = currVolume;
}
if(height[left] > height[right])
{
right--;// 右指针左移
}
else
{
left++;// 左指针右移
}
}
return maxVolume;
}
};