题目链接
https://leetcode-cn.com/problems/container-with-most-water/
题目
给你 n 个非负整数 a1,a2,...,an,每个数代表坐标中的一个点 (i, ai) 。在坐标内画 n 条垂直线,垂直线 i 的两个端点分别为 (i, ai) 和 (i, 0) 。找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
说明:你不能倾斜容器。
(该题为字节跳动等经典面试题)
示例
示例 1:
输入:[1,8,6,2,5,4,8,3,7]
输出:49
解释:图中垂直线代表输入数组 [1,8,6,2,5,4,8,3,7]。在此情况下,容器能够容纳水(表示为蓝色部分)的最大值为 49。示例 2:
输入:height = [1,1]
输出:1示例 3:
输入:height = [4,3,2,1,4]
输出:16示例 4:
输入:height = [1,2,1]
输出:2
提示
思路及代码
很容易想到,最简单的方法,双层for暴力枚举:
class Solution {
public:
int maxArea(vector<int>& height) {
int water=0;
int n=height.size();
for(int i=0;i<n-1;i++)
{
for(int j=i+1;j<n;j++)
{
water=max( water, (j-i)*min(height[i],height[j]) );
}
}
return water;
}
};
暴力超时,gg
于是改用双指针,left指向最左边,righjt指向最右边,如果左边的高度更低则左边的指针右移,如果右边的高度更低则右边的指针左移,每次缩小底部长度,以寻找更大的高度,使得整个容器面积更大,直到左边的指针遇到右边的指针结束,代码如下:
class Solution {
public:
int maxArea(vector<int>& height) {
int water=0;
int result=0;
int left=0;
int right=height.size()-1;
int h=0;
while(left<right)
{
h=min(height[left],height[right]);
water=(right-left)*h;
result=max(result,water);
if(height[left]<height[right])
{
left++;
}
else
{
right--;
}
}
return result;
}
};