弟中弟的Leetcode总结——数组类(二)
题目描述
Container With Most Water
Given n non-negative integers a1, a2, …, an , where each represents a point at coordinate (i, ai). n vertical lines are drawn such that the two endpoints of line i is at (i, ai) and (i, 0). Find two lines, which together with x-axis forms a container, such that the container contains the most water.
Note: You may not slant the container and n is at least 2.
Example:
Input: [1,8,6,2,5,4,8,3,7]
Output: 49
思路——Naive Solution
最简单的方法就是暴力枚举,从第一个开始一直到最后一个,算他们的乘积。
代码(py3)
class Solution:
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
leng=len(height)
maxv=0
for i in range(leng):
for j in range(i+1,leng):
val=min(height[i],height[j])*(j-i)
maxv=val if(val>maxv) else maxv
return maxv
思路——Improved Solution
前面的解法时间为O(nlog(n)),显然会超时,因此需要对算法进行改进。
之前的算法是从左到右逐一扫描,对于每一个数,都需要扫描(n-i)遍。但是在这之中忽略了给定的数值的作用。显然如果数值越高,
那么能装的也就越多。因此可以设定两个指针,开始的时候指向左右两端,然后逐渐向中间移动,每次移动的时候移动数值较小的那个。
代码(py3)
class Solution:
def maxArea(self, height):
"""
:type height: List[int]
:rtype: int
"""
leng=len(height)
maxv=0
i=0
j=leng-1
while(i!=j):
val=min(height[i],height[j])*(j-i)
maxv=val if(val>=maxv) else maxv
if(height[i]<height[j]):
i+=1
else:
j-=1
return maxv