974. 和可被 K 整除的子数组

974. 和可被 K 整除的子数组

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

思路：利用前缀和的思想，p[i] = A[0] + A[1] + … + A[i], 则每个连续子数组的和sum(i,j)
就可以写成 p(j) - p(i-1),判断子数组能否被k整除 相当于判断 (p(j) - p(i-1) % k == 0 ？

(用p(i)表示数组中0到i数字的总和，如果p(j)与p(i)对k取余得到的结果是相同的， 也就是p(j) -p(i) = nK ( n>0 )，那么表示数组 i 到 j 的元素能构成一个这样的子数组)

根据同余定理 (p(j) - p(i-1) mod k == 0 等价于 p(j) mod k == p(i-1) mod k

然后对数组进行遍历，在遍历同时统计答案。当我们遍历到第 i 个元素时，统计以 i 结尾的符合条件的子数组个数。我们可以维护一个以前缀和模 k 的值为键，出现次数为值的哈希表map，在遍历的同时进行更新。这样在计算以 i 结尾的符合条件的子数组个数时，答案即为map[p(i)mod k]

最后的答案即为以每一个位置为数尾的符合条件的子数组个数之和。需要注意的一个边界条件是，我们需要对哈希表初始化map[0] = 1，这样就考虑了前缀和本身被 k 整除的情况。

class Solution {
public:
    int subarraysDivByK(vector<int>& nums, int k) {
        int n = nums.size();
        unordered_map<int,int>map = {{0,1}};
        int sum = 0;
        int count = 0;
        int div;
        for( auto num:nums)
        {
            sum += num;
            div = ((sum % k)+ k)%k;//负数情况
            if(map.count(div))
            {
                count +=map[div];
            }
            ++map[div];
        }
        return count;

    }
};

时间复杂度：O(N）
空间复杂度：O(min(n,k))