【JavaScript】配列の各サブ配列の合計を計算して新しい配列を返す方法

問題の概要

数値の集合を含む複数の配列（ネストされた配列）が与えられたとき、対応する各サブ配列内のすべての数値の合計を要素とする配列を返す関数を作成します。

例

たとえば、入力配列が次のようになっているとします。

const numbers = [
    [1, 2, 3, 4],
    [5, 6, 7],
    [8, 9, 10, 11, 12]
];

この場合、関数の出力は次のようになるはずです。

const output = [10, 18, 50];

実装方法

この問題は、Array.prototype.reduce() メソッドを組み合わせることで簡潔に解決できます。まず、1つの配列の合計を計算する関数を定義し、それを各サブ配列に順番に適用していきます。

const numbers = [
    [1, 2, 3, 4],
    [5, 6, 7],
    [8, 9, 10, 11, 12]
];
const sum = arr => arr.reduce((acc, val) => acc+val);
const sumSubArray = arr => {
    return arr.reduce((acc, val) => {
        const s = sum(val);
        acc.push(s);
        return acc;
    }, []);
};
console.log(sumSubArray(numbers));

コードの解説

まず、sum 関数は reduce() を使って、渡された配列内の全要素を合計します。次に、sumSubArray 関数は外側の配列に対して reduce() を実行し、初期値として空の配列を用意します。各サブ配列に対して sum 関数を呼び出して合計を求め、その結果を結果配列に追加していきます。最終的に、すべてのサブ配列の合計を格納した配列が返されます。

出力

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

[ 10, 18, 50 ]