JavaScriptで配列内の要素から目標の合計値を持つグループを数える方法

問題の概要

まず、次のような数値の配列を例に考えてみましょう。

const arr = [1, 2, 1, 3, 2];

ここで求められているのは、次の3つの引数を受け取るJavaScript関数です。

• 第1引数：対象となる数値の配列
• 第2引数：目標となる合計値（sum）
• 第3引数：合計値を構成する要素の個数（num）※同じ要素の重複使用は不可

関数は、指定された個数の要素の合計が目標値と一致するグループの総数を返します。

入力例と出力例

入力値が次のとおりだとします。

const arr = [1, 2, 1, 3, 2];
const sum = 3;
const num = 2;

この場合、期待される出力は次のとおりです。

const output = 2;

これは、合計が3になる連続する2要素のグループが「[1, 2]」と「[2, 1]」の2つ存在するためです。

サンプルコード

この問題を解くコードは次のとおりです。

const arr = [1, 2, 1, 3, 2];
const sum = 3;
const num = 2;
const findGroups = (arr = [], sum = 1, num = 1) => {
    let count = 0
    for(let i = 0; i < arr.length; i++){
        let part = arr.slice(0 + i, num + i);
        const partSum = part.reduce((acc, val) => acc + val);
        if(partSum === sum){
            count++;
        };
    };
    return count
};
console.log(findGroups(arr, sum, num));

コードの解説

このコードでは、スライディングウィンドウの考え方を利用しています。処理の流れは以下のとおりです。

1. 条件を満たすグループの数を記録するため、変数countを0で初期化します。
2. forループで配列の先頭から順に走査し、slice()メソッドでnum個分の連続する要素を取り出します。
3. reduce()メソッドで取り出した要素の合計値を計算し、目標のsumと一致するかを判定します。
4. 一致した場合はcountを1増やします。
5. ループ完了後、countを返り値として返します。

なお、より堅牢な実装にする場合は、条件をif(part.length === num && partSum === sum)のように要素数のチェックも加えると、配列の末尾で要素数が不足するケースを確実に除外できます。

実行結果

コンソールには次のように出力されます。

2