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

题目:给定一个整数数组 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));

你可能感兴趣的:(算法,javascript,leetcode)