题目评论中置顶的几个都是采用O(n)的解法,不得不说确实想法是妙,自己倒没往那个方面想到。
O(N)的思路是:一开始两个指针一个指向开头一个指向结尾,此时容器的底是最大的,接下来随着指针向内移动,会造成容器的底变小,在这种情况下想要让容器盛水变多,就只有在容器的高上下功夫。 那我们该如何决策哪个指针移动呢?我们能够发现不管是左指针向右移动一位,还是右指针向左移动一位,容器的底都是一样的,都比原来减少了 1。这种情况下我们想要让指针移动后的容器面积增大,就要使移动后的容器的高尽量大,所以我们选择指针所指的高较小的那个指针进行移动,这样我们就保留了容器较高的那条边,放弃了较小的那条边,以获得有更高的边的机会。
再说下我的思路,假如说例子为:
4,3,2,1,4
那么我先根据height从大到小排序
高度y:4,4,3,2,1
索引x:0,4,1,2,3
对于第i个(x,y)作为一条边,如果需要装下高度为y的水,另一边肯定是位于它左边高于它的。
那么剩下的就是水的宽度了,需要记录下0~i-1中索引的最小值minx和maxx。
那么对于第i个坐标,装水的高度>=y的最大容积,另一边肯定是minx和maxx中的其中一条。
从i=1开始遍历,取最大的容积即可。
由于需要排序,因此复杂度为O(NlogN)
执行用时:32 ms, 在所有 C++ 提交中击败了65.41%的用户
内存消耗:17.5 MB, 在所有 C++ 提交中击败了61.58%的用户
class Solution {
public:
struct Node{
int y, x;
bool operator<(const Node tmp) const{
return y>tmp.y; //按高度从大到小排序
}
};
int maxArea(vector<int>& height) {
int len=height.size();
Node inputs[len+1];
for(int i=0;i<len;i++){
Node tmp;
tmp.y=height[i];
tmp.x=i;
inputs[i]=tmp;
}
sort(inputs,inputs+len);
int minx = inputs[0].x;
int maxx = inputs[0].x;
int ans=0;
for(int i=1;i<len;i++){
int a=abs(inputs[i].x-minx) * inputs[i].y;
int b=abs(inputs[i].x-maxx) * inputs[i].y;
ans=max(max(a,b),ans);
minx=min(inputs[i].x,minx);
maxx=max(inputs[i].x,maxx);
}
return ans;
}
};