题目描述
给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。返回容器可以储存的最大水量。
测试用例
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。
解题思路
1.错误思路之暴力拆解-----超时
直接计算所有的垂线和其他垂线的面积的大小,然后进行比较,时间复杂度为0(n*n)
2.正确思路之双指针法
使用两个指针,一个指向数组的头部,一个指向数组的尾部,计算两个指针之间指向的面积,然后移动高度低的那个指针,直到两个指针相等
原理:计算最大面积时,且长和高都为变量时,可以考虑向固定一个变量,先将宽固定为最大值,然后无论是向左移动一格还是向右移动一格,宽都是减少1,而为了使面积最大,我们保留高的那个垂线,移动低的那个指针。
代码实现
#include <iostream>
#include<algorithm>
#include <vector>
#include<stack>
#include <cctype>
#include <stdio.h>
#include <iomanip>
using namespace std;
//暴力搜索法:直接计算所有的垂线和其他垂线的面积的大小,然后进行比较,时间复杂度为0(n*n)--超时
// 双指针法:设置两个指针,一个指向数组的首部,一个指向数组的尾部,始终移动数组中指向小的那个指针
int maxArea(vector<int>& height)
{
int maxArea1=0;
int Area=0;
vector<int>::iterator it1=height.begin();
vector<int>::iterator it2=height.end()-1;
while(it1!=it2){
//distance(first, last)函数会返回[first, last)范围内包含的元素个数
Area=(min(*it1,*it2))*(distance(it1,it2));
if( Area>maxArea1){
maxArea1=Area;
}
if(*it1>*it2)
*it2--;
else{
*it1++;
}
}
return maxArea1;
}
int main()
{
vector<int> b;
b.push_back(1);
b.push_back(8);
b.push_back(6);
b.push_back(2);
b.push_back(5);
b.push_back(4);
b.push_back(8);
b.push_back(3);
b.push_back(7);
cout<<maxArea(b);
}