牛客题霸 未排序数组中累加和为给定值的最长子数组长度

题目描述

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

 

示例1

输入

[1,-2,1,1,1],0

返回值

3

 

解题思路：

为了解答题目，引入一个概念，s(i)代表子数组arr[0..i]所有元素的累加和。那么子数组arr[j-1, i](0<=j<=i<arr.length)的累加和为s(i)-s(j-1)。

1. 设置变量sum=0，表示从0位置开始一直加到i位置所有元素的和。设置变量len＝0，表示累加和为k的最长子数组长度。定义一个HashMap，其中key是sum值，value为sum值最早出现的位置。

2. 从左到右开始遍历，当前元素为arr[i]

1) sum = sum + arr[i]，即s(i)，在map中查看是否存在sum-k

    如果sum-k存在，从map中取出sum-k对应的value，记为j，得到s(i)-s(j)=k，所以此时arr[j+1, i]的长度即为题目要求的子数组的长度，如果长度大于len，则更新len。

2) 检查当前的sum(即s(i))是否在map中，若不存在说明是第一次出现，就把记录加入到map中。

* 注意 根据arr[j+1, i]的累加和为s(i)-s(j)，所以，如果从0开始累加，则j+1>=1。意味着，所有从0开始的子数组都没有考虑过，所以，应该从-1位置开始累加，也就是在遍历之前，先把(0,-1)放进map，也就是如果任何一个数都不加时，累加和为0。

 

Python代码：

#
# max length of the subarray sum = k
# @param arr int整型一维数组 the array
# @param k int整型 target
# @return int整型
#
class Solution:
    def maxlenEqualK(self , arr , k ):
        # write code here
        if not arr:
            return 0
        data = {}
        data[0] = -1
        res = 0
        s = 0
        for i, num in enumerate(arr):
            s += num
            if s not in data:
                data[s] = i
            if s-k in data:
                res = max(res, i - data[s-k])
        return res