题目:给定一个整数数组 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]
var subarraysDivByK = function(nums, k) {
//变量
let sum=0;let key;let ans=0;let value;
let map=new Map();map.set(0,1);//特殊点,有余数=0,即代表它从该地点到头整个连续和能够整除k
//遍历数组
for(let i=0;i<nums.length;i++){
//到该i值时的前缀和
sum+=nums[i];
//每个此时前缀和(包括负数)对k的余数
key=sum%k;//数组对k的余数作为散列表的键
if(key<0){key+=k;}//sum是负数对k的余数
//将余数值和出现该余数的次数加入到散列表中
if(map.has(key)){//如果该键没有存在过,那么值直接为1,如果存在过,值++
value=map.get(key);
map.set(key,value+1);
}else{
map.set(key,1);
}
}
//遍历散列表
for(let n of map.values()){
ans+=n*(n-1)/2;
}
return ans;
};
console.log(subarraysDivByK([4,5,0,-2,-3,1],5));