题目
给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 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
提示:
- n == height.length
- 2 <= n <= 105
- 0 <= height[i] <= 104
题解
class Solution {
public:
int maxArea(vector<int>& height) {
int max = (height.size()-1)*min(height[0],height[height.size()-1]);
int left = 0, right = height.size() - 1;
while(left<right){
if(height[left]<height[right]){
left++;
if(height[left]>height[left-1] && max < min(height[left], height[right])*(right-left)){
max = min(height[left], height[right])*(right-left);
}
}
else{
right--;
if(height[right] > height[right+1] && max < min(height[left], height[right])*(right-left)){
max = min(height[left], height[right])*(right-left);
}
}
}
// for(int i=0;i<height.size();i++){
// for(int j=i+1;j<height.size();j++){
// int area = (j-i)*min(height[j],height[i]);
// if( area > max ){
// max = area;
// }
// }
// }
return max;
}
};
笔记
- 先试了试暴力法,果然是超时的;
- 双指针遍历一遍数组,时间复杂度O(n);
- 同样是剪枝判断,对那些确定不会增加容积的行为:比如向内移动长板,不再参与运算。由此分析,就是单边移动。
- 关键点就在于:
- 要确定从两端开始选择,此时底边是最大的,只需在底边变小的趋势下,考虑高度变化;
- 基于短板效应,只有短边的移动是有意义的;
- 底下是精选答案,简洁优雅:
class Solution {
public:
int maxArea(vector<int>& height) {
int i = 0, j = height.size() - 1, res = 0;
while(i < j) {
res = height[i] < height[j] ?
max(res, (j - i) * height[i++]):
max(res, (j - i) * height[j--]);
}
return res;
}
};
/*
作者:jyd
链接:https://leetcode.cn/problems/container-with-most-water/solution/container-with-most-water-shuang-zhi-zhen-fa-yi-do/
来源:力扣(LeetCode)
著作权归作者所有。商业转载请联系作者获得授权,非商业转载请注明出处。
*/