可被 n 整除之和 (JavaScript)
問題
我們需要編寫一個 JavaScript 函式,該函式將一個數字陣列 arr 作為第一個引數,將一個數字 num 作為第二個引數。我們的函式應返回具有可被 num 整除的總和的子陣列數量(連續且非空)。
例如,如果輸入的函式為 −
const arr = [4, 5, 0, -2, -3, 1]; const num = 5;
則輸出應為 −
const output = 7;
輸出說明
有 7 個子陣列的總和可被 5 整除 −
[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
廣告
資料結構
網路
關係型資料庫管理系統
作業系統
Java
iOS
HTML
CSS
Android
Python
C 程式設計
C++
C#
MongoDB
MySQL
JavaScript
PHP