20、和可被 K 整除的子数组

题目：

给定一个整数数组 A，返回其中元素之和可被 K 整除的（连续、非空）子数组的数目。

示例：

输入：A = [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 (A, K) {
  let map = new Map()
  let sum = 0
  for (let n of A) {
    sum += n
    const key = (sum % K + K) % K
    map.set(key, map.has(key) ? map.get(key) + 1 : 1)
  }
  let result = map.has(0) ? map.get(0) : 0
  for (let [, val] of map) {
    if (val > 1) result += val * (val - 1) / 2
  }
    console.log(result);
  return result;
};
A = [4,5,0,-2,-3,1]; K = 5;
subarraysDivByK(A, K);

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/subarray-sums-divisible-by-k
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。