Java怎么找到和为K的子数组

本篇内容主要讲解“Java怎么找到和为K的子数组”,感兴趣的朋友不妨来看看。本文介绍的方法操作简单快捷,实用性强。下面就让小编来带大家学习“Java怎么找到和为K的子数组”吧!

 Java怎么找到和为K的子数组

前序和

定义sum[i]表示从开始加到第i位置的和,在数组[1,2,3,4]上我们就有

sum[0] = nums[0]

sum[1] = nums[0] + nums[1] = sum[0] + nums[1]

sum[2] = nums[0] + nums[1] + nums[2] = sum[0] + nums[1] + nums[2] = sum[1] + nums[2]

sum[3] = nums[0] + nums[1] + nums[2] +nums[3] = sum[0] + nums[1] + nums[2] + nums[3] = ... ...

我们发现,我们做了很多重复的计算,这些就是我们可以优化的地方,对上面第3个式子进行一下变形,

sum[1]-sum[0]=nums[1]

sum[2] - sum[0] = nums[1] + nums[2]

sum[2] - sum[1] = nums[2]

sum[i] - sum[j] 就是(j+1,i)之间连续数组的和,如果sum[i] - sum[j] = k,那我们就可以进行计数了,这里也可以进行一次变形sum[j] = sum[i] - k,我们使用哈希表来记录sum[j],再次出现的时候我们就进行计数。

func subarraySum(nums []int, k int) int {    res,sum,m :=0,0, map[int]int{0:1}    for i:=0;i<len(nums);i++{        sum += nums[i]        if m[sum-k]>0{            res += m[sum-k]        }        m[sum]++    }    return res}

到此,相信大家对“Java怎么找到和为K的子数组”有了更深的了解,不妨来实际操作一番吧!这里是亿速云网站,更多相关内容可以进入相关频道进行查询,关注我们,继续学习!