（算法）和可被K整除的⼦数组————＜前缀和数组＞

1. 题⽬链接：974.和可被K整除的⼦数组

2. 题⽬描述：

3. 解法（前缀和在哈希表中）：

（暴⼒解法就是枚举出所有的⼦数组的和，这⾥不再赘述。）

本题需要的前置知识：

• 同余定理 如果(a - b) % n == 0 ，那么我们可以得到⼀个结论： a % n == b % n 。⽤⽂字叙 述就是，如果两个数相减的差能被n整除，那么这两个数对n取模的结果相同。

例如： (26 - 2) % 12 == 0 ，那么26 % 12 == 2 % 12 == 2 。 

• c++ 中负数取模的结果，以及如何修正「负数取模」的结果 

        a. c++ 中关于负数的取模运算，结果是「把负数当成正数，取模之后的结果加上⼀个负号」。 

                例如： -1 % 3 = -(1 % 3) = -1 b. 因为有负数，为了防⽌发⽣「出现负数」的结果，以(a % n + n) % n 的形式输出保证为 正。 

                例如： -1 % 3 = (-1 % 3 + 3) % 3 = 2

算法思路：

思路与560.和为K的⼦数组这道题的思路相似。

设i 为数组中的任意位置，⽤sum[i] 表⽰[0, i] 区间内所有元素的和。 

• 想知道有多少个「以i 为结尾的可被k 整除的⼦数组」，就要找到有多少个起始位置为x1, x2, x3... 使得[x, i] 区间内的所有元素的和可被k 整除。 

• 设[0, x - 1] 区间内所有元素之和等于a ， [0, i] 区间内所有元素的和等于b ，可得 (b - a) % k == 0 。

• 由同余定理可得， [0, x - 1] 区间与[0, i] 区间内的前缀和同余。于是问题就变成： 

        ◦ 找到在[0, i - 1] 区间内，有多少前缀和的余数等于sum[i] % k 的即可。 我们不⽤真的初始化⼀个前缀和数组，因为我们只关⼼在i 位置之前，有多少个前缀和等于 sum[i] - k 。

因此，我们仅需⽤⼀个哈希表，⼀边求当前位置的前缀和，⼀边存下之前每⼀种前 缀和出现的次数。

C++算法代码：

class Solution 
{
public:
    int subarraysDivByK(vector<int>& nums, int k) 
    {
        //记录余数和出现次数
        unordered_map<int,int>hash;
        hash[0%k]=1;
        int sum=0,ret=0;
        for(auto x:nums)
        {
            sum+=x;
            int r=(sum%k+k)%k;  //余数保证为正数
            //如果现在的余数前缀和==之前点的余数前缀和
            if(hash.count(r))
            {
                //因为sum在变，故使用+=
                ret+=hash[r];
            }
            hash[r]++;  //记录余数
        }
        return ret;
    }
};

 Java算法代码：

class Solution {
	public int subarraysDivByK(int[] nums, int k) {
		Map<Integer, Integer> hash = new HashMap<Integer, Integer>();
		hash.put(0 % k, 1);
		int sum = 0, ret = 0;
		for (int x : nums)
		{
			sum += x; // 计算当前位置的前缀和 
			int r = (sum % k + k) % k;
			ret += hash.getOrDefault(r, 0); // 统计结果 
			hash.put(r, hash.getOrDefault(r, 0) + 1);
		}
		return ret;
	}
}