JavaScriptでnで割り切れる合計

問題

最初の引数として数値の配列arrを取り、2番目の引数として数値numを受け取るJavaScript関数を作成する必要があります。この関数は、合計がnumで割り切れる（連続した、空でない）サブ配列の数を返す必要があります。

たとえば、関数への入力が-

const arr = [4, 5, 0, -2, -3, 1];
const num = 5;

その場合、出力は-

const output = 7;

出力の説明

合計が5で割り切れる7つのサブ配列があります-

[4, 5, 0, -2, -3, 1], [5], [5, 0], [5, 0, -2, -3], [0], [0, -2, -3], [-2, -3]

例

このためのコードは-

const arr = [4, 5, 0, -2, -3, 1];
const num = 5;
const divisibleSum = (arr = [], num = 1) => {
   const map = {};
   let sum = 0;
   for (let i = 0; i < arr.length; i++) {
      sum += arr[i];
      const key = ((sum % num) + num) % num;
      map[key] = map[key]+1||1;
   };
   let s = 0;
   for (let i = 0; i < num; i++) {
      if (map[i] > 1) {
         s += (map[i] * (map[i] - 1)) / 2;
      }
   }
   return s + (map[0]||0);
};
console.log(divisibleSum(arr, num));

出力

そして、コンソールの出力は-

7