11. 盛最多水的容器
题目:给定一个长度为 n 的整数数组 height 。有 n 条垂线,第 i 条线的两个端点是 (i, 0) 和 (i, height[i]) 。
找出其中的两条线,使得它们与 x 轴共同构成的容器可以容纳最多的水。
返回容器可以储存的最大水量。
链接:https://leetcode.cn/problems/container-with-most-water
个人思路
- 暴力解法:两次for循环把每个可能的面积都算出来
class Solution:
def maxArea(self, height: List[int]) -> int:
ans = 0
for i in range(0,len(height)):
for j in range(i+1,len(height)):
border = min(height[i],height[j])
ans = max(ans,border*(j-i))
return ans
但超时了
2. 改进暴力解法:一个从前面往后循环,一个从后面往前循环,当后面循环的数比前面要大时,说明后面的再往前循环也毫无意义,故break
class Solution:
def maxArea(self, height: List[int]) -> int:
ans = 0
for i in range(0,len(height)):
for j in range(len(height)-1,i,-1):
if height[j] >= height[i]:
ans = max(ans,height[i]*(j-i))
break
else:
border = min(height[i],height[j])
ans = max(ans,border*(j-i))
return ans
还是超时了
3. 继续改进,当前面循环的比之前循环的数要小时,继续循环(这里其实已经挺接近答案的双指针解法了)
class Solution:
def maxArea(self, height: List[int]) -> int:
ans = 0
temp = 0
for i in range(0,len(height)):
if temp > height[i]:
continue
temp = height[i]
for j in range(len(height)-1,i,-1):
if height[j] >= height[i]:
ans = max(ans,height[i]*(j-i))
break
else:
border = min(height[i],height[j])
ans = max(ans,border*(j-i))
return ans
还是超时了
- 双指针:自己本来想到了双指针,但没有动力去证明可行性:
class Solution:
def maxArea(self, height: List[int]) -> int:
l, r = 0, len(height) - 1
ans = 0
while l < r:
area = min(height[l], height[r]) * (r - l)
ans = max(ans, area)
if height[l] <= height[r]:
l += 1
else:
r -= 1
return ans
复杂度分析
时间复杂度:O(N),双指针总计最多遍历整个数组一次。
空间复杂度:O(1),只需要额外的常数级别的空间。
证明其实也不难,可参考
链接:https://leetcode.cn/problems/container-with-most-water/solution/sheng-zui-duo-shui-de-rong-qi-by-leetcode-solution/