【题目】
给定一个数组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].
【代码实现】
#python3.5
def getMaxLength(arr, k):
if arr == None or len(arr) == 0 or k < 1:
return 0
left = 0
right = 0
length = 0
sum = arr[0]
while left < len(arr) and right < len(arr):
if sum == k:
length = max(length, right-left+1)
sum -= arr[left]
left += 1
elif sum > k:
sum -= arr[left]
left -= 1
else:
right += 1
if right == len(arr):
break
sum += arr[right]
return length