找到给定总和的子数组 |第 1 组(非负数)

在程序开发中，有时需要找出一个数组中，元素总和为给定值的所有子数组。本文将介绍如何解决这个问题。假设数组中元素均为非负整数。

思路

我们可以使用双指针的方法来实现。定义两个指针left和right，分别指向子数组的左右两端。初始时，left和right均指向数组的第一个元素。我们累计[left,right]中所有元素的和sum，若sum不大于给定值，则右移right指针；若sum大于给定值，则左移left指针。当sum等于给定值时，我们就找到了一个符合条件的子数组，并将其加入结果数组中。然后我们将left指针右移一位，继续寻找下一个符合条件的子数组。

代码

def find_subarrays(arr, target):
    n = len(arr)
    res = []
    left = right = sum = 0
    
    while right < n:
        sum += arr[right]
        
        while left <= right and sum > target:
            sum -= arr[left]
            left += 1
        
        if sum == target:
            res.append(arr[left:right+1])
        
        right += 1
    
    return res

参数说明

• arr：待查找子数组的数组
• target：给定的元素总和

示例

>>> arr = [1,2,3,4,5]
>>> target = 9
>>> find_subarrays(arr, target)
[[2, 3, 4], [4, 5]]

以上代码片段返回了原数组[1,2,3,4,5]中，元素总和等于9的所有子数组[[2, 3, 4], [4, 5]]。