给定一个整数数组 nums 和一个整数 k ,返回其中元素之和可被 k 整除的(连续、非空) 子数组 的数目。
子数组 是数组的 连续 部分。
示例 1:
输入:nums = [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]
示例 2:输入: nums = [5], k = 9
输出: 0提示:
1 <= nums.length <= 3 * 10^4
-104 <= nums[i] <= 104 2 <= k <= 10^4来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/subarray-sums-divisible-by-k
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
数据量为1 <= nums.length <= 3 * 10^4,直接使用前缀和+差分
时间复杂度O(n^2),会超时。
当i <= j,S[i] - S[ j - 1] 表示前缀和 a[ j ] + a[ j + 1 ] + … a[ i ]
如果前缀和是K的倍数:
则 ( S[i] - S[ j - 1] ) mod k == 0
根据同余定理:
得到 ( S[i] ) mod k - ( S[ j - 1] ) mod k == 0
得到 ( S[i] ) mod k == ( S[ j - 1] ) mod k
所以只要在前缀和数组中统计 t = ( S[i] ) mod k 的个数,相同t可以两两组合,利用组合公式算出该t的所有组合,再将不同t的数量相加即可。
其中m = 2, n等于每个t内的数量
C n m = n ! / (2 ! * (n - 2) ! ) = n * (n - 1) / 2
class Solution {
public int subarraysDivByK(int[] nums, int k) {
int[] map = new int[k];
int p = 0;
int[] prefixSum = new int[nums.length + 1];
for (int i = 1; i < prefixSum.length; i++) {
prefixSum[i] = p + nums[i - 1];
p = prefixSum[i];
}
for (int i = 0; i < prefixSum.length; i++) {
map[ prefixSum[i] % k < 0 ? (prefixSum[i] % k + k) :
prefixSum[i] % k ]++;
}
int ans = 0;
for (int i = 0; i < map.length; i++) {
ans += map[i] * (map[i] - 1) / 2;
}
return ans;
}
}