题目:给定一个数组arr,该数组无序,但每个值都为正数,在给定一个正数k。求arr中所有的子数组中所有元素累加为k的最长子数组长度。
例如,arr = [1,2,1,1,1],k = 3.
累加和为3的最长子数组为[1,1,1],所以返回结果为3。
基本思路:
使用两个指针left和right,代表子数组的范围,初始时都为 0。使用变量sum记录子数组的累加和,初始为arr[0]。即arr[0…0]的累加和。根据sum与k的比较结果确定指针的移动:
1、如果 sum == k,说明目前子数组的累加和满足条件,该子数组的长度可知 right - left + 1。因为整个数组元素都是正数,所以,在扩展该子数组显然不可能等于k,所以我们应该令left加1,开始考察以 left + 1 位置开始的子数组,同时令 sum -= arr[left].
2、如果 sum < k,说明还需要加上 right 后面的元素,所以令right + 1,同时令 sum += arr[right+1]。这里需要注意 right + 1 后是否越界。
3、如果 sum > k,说明此时子数组的累计和已经大于k,所以令left + 1 表示开始考虑以 left + 1 开始的子数组,同时令 sum -= arr[left].
def getMaxLength(L,k):
if L == None or len(L)<1 or k < 0:
return 0
left = 0
right = 0
sum_ = L[0]
length = len(L)
while right < len(L):
if sum_ == k:
length = max(length,right-left+1)
sum_ -= L[left]
left +=1
elif sum_ < k:
right = right +1
if right == len(L):
break
sum_ += L[right]
else:
sum_ -= L[left]
left +=1
return length