和可被K整除的子数组

给定一个整数数组A，返回其中元素之和可被K 整除的（连续、非空）子数组的数目。

示例：

输入：A = [4,5,0,-2,-3,1], K = 5
输出：7
解释：
有 7 个子数组满足其元素之和可被 K = 5 整除：
[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]

提示：

1 <= A.length <= 30000-10000 <= A[i] <= 100002 <= K <= 10000 

思路：

这道题刚接到手

不用想肯定是用前缀和

题目都说了嘛，子数组元素之和

我们可以先获取前缀和的数组，比如[4,5,0,-2,-3,1]

前缀和数组：[0，4，9，9，7，4，5]

数组第一个必定是0，不然不好算第一个元素的前缀和

现在问题变成了，preSum[j+1]-preSum[i]能被K整除，数组范围是[i…j]

那么能被K整除，就说明两个前缀和求余K是相等的

由:（preSum[j+1]-preSum[i]）%K=0

得：preSum[j+1]%K-preSum[i]%K=0

得：preSum[j+1]%K=preSum[i]%K

对上述前缀和求余

得到：[2,0,1,0,4]

其中2代表[0,5]是求余为0的个数

1代表[7]求余为2的个数

4代表[4,9,9,4]求余为4的个数

数组下标就是余数，元素就是求余相同的个数

特别注意：取余是负数时，需要加上K，因为数组下标不能是负数

那么如何求被K整除的子数组的个数呢？

比如，余数为4的前缀和有4个

那么我们只需要求它们的排列组合就行了，即：4*(4-1)/2

代码:

//前缀和
public int subarraysDivByK(int[] A, int K) {
    int res=0;
    int[] preSum=new int[A.length+1];
    preSum[0]=0;
    //计算前缀和
    for(int i=1;i<preSum.length;i++)preSum[i]=preSum[i-1]+A[i-1];
    int[] cnt=new int[K];
    for(int pre:preSum)cnt[pre%K<0?pre%K+K:pre%K]++;
    for(int c:cnt)
        res+=c*(c-1)/2;

    return res;

}

复杂度：

时间复杂度：O(n)

空间复杂度：O(n+K)