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

最新推荐文章于 2022-01-13 00:14:33 发布

孤舟钓客

最新推荐文章于 2022-01-13 00:14:33 发布

阅读量356

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/guzhou_diaoke/article/details/103838213

版权

算法专栏收录该内容

100 篇文章 1 订阅

订阅专栏

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

题目

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

思路

构建一个映射关系map[sum] = index, 表示[0, index] 子数组的累加和为sum，且index是sum最早出现的位置。
从头遍历数组并求累加和，假设当前累加和为current，假如current-k在map中，索引为index，表示到index位置累加和为current-k，则从index+1到当前位置累加和为k。
[0, …, index][index+1, …, i]
则找到一个长度为i-index的子数组和为k。

实现

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

    sum_map = {0: -1}
    current = 0
    ans = 0

    for i in range(len(arr)):
        current += arr[i]
        if current-k in sum_map:
            ans = max(ans, i-sum_map[current-k])
        if current not in sum_map:
            sum_map[current] = i

    return ans

测试

使用暴力搜索做验证：

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

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

    return ans


def test(count, maxval):
    arr = [random.randint(-maxval, maxval) for _ in range(count)]
    total = abs(sum(arr))
    for k in range(-total, total+1):
        ans1 = max_length_of_sub_array_with_sum(arr, k)
        ans2 = max_length_of_sub_array_with_sum2(arr, k)
        assert(ans1 == ans2)


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