00前缀和中等 LeetCode974. 和可被 K 整除的子数组

分析

前缀和，涉及子数组和的问题首先想到前缀和能不能解决。
要想子数组的和能整除k,要是这两个对应的前缀和的差求k的余数为0，即(a-b)%k == 0 >> (a%k - b%k) %k == 0
另外被除数是负数去求余结果还是负数，需要加上除数K将它转成正数。
前缀和差求k的余数为0，只需要两个前缀和对k的余数相等即可。
分别求出余数是0~k-1的各个前缀和的个数，然后计算同一个余数下所有前缀和做差的情况，使得差对k的余数为0。余数是0的情况要，不需要两个前缀和，单个前缀和也是能让子数组和整除k。
若前缀和是0需要特殊处理，给余数还要再加上k。

class Solution {
    public int subarraysDivByK(int[] nums, int k) {
int n = nums.length;
int[] preSum = new int[n];
int[] mo = new int[k];
for (int i = 0; i < n; i++) {
	if (i == 0) {
		preSum[i] = nums[0];
	} else {
		preSum[i] = preSum[i-1] + nums[i];
	}
	int num = preSum[i] % k;
	if (num < 0) {
		num += k;
	}
	mo[num]++;
}
int ans = mo[0];
for (int i = 0; i < k; i++) {
	ans += mo[i] * (mo[i] - 1) / 2;
}
return ans;
    }
}