未排序正数数组中累加和为给定值的最长子数组长度

未排序正数数组中累加和为给定值的最长子数组长度

题目

给定一个无序正整数数组arr及一个正数k，求arr的所有子数组中元素相加和为k的最长子数组长度。

思路

两个指针left, right，可以表示子数组[left, right], current是子数组的和，

• 若current小于k，right右移
• 若current大于k，left左移
• 若current等于k，更新结果

def max_lenght_of_sub_array_with_sum1(arr, k):
    if arr is None or len(arr) == 0 or k <= 0:
        return 0

    left, right = 0, 0
    current = arr[0]
    ans = 0

    while right < len(arr):
        if current == k:
            ans = max(ans, right-left+1)
            current -= arr[left]
            left += 1
        elif current < k:
            right += 1
            if right != len(arr):
                current += arr[right]
        else:
            current -= arr[left]
            left += 1

    return ans

测试

用暴力搜索法做验证：

import random


def max_lenght_of_sub_array_with_sum0(arr, k):
    if arr is None or len(arr) == 0 or k <= 0:
        return 0

    ans = 0
    for i in range(len(arr)):
        for j in range(i, len(arr)):
            if k == sum(arr[i:j+1]):
                ans = max(ans, j-i+1)

    return ans

def test(count, maxval):
    arr = [random.randint(1, maxval) for _ in range(count)]
    total = sum(arr)
    for k in range(total+1):
        ans0 = max_lenght_of_sub_array_with_sum0(arr, k)
        ans1 = max_lenght_of_sub_array_with_sum1(arr, k)
        assert(ans0 == ans1)


if __name__ == '__main__':
    test(10, 10)
    test(10, 100)
    test(100, 10)