JavaScriptで配列の累積合計を求める方法

たとえば、次のような数値の配列があるとします。

const arr = [1, 2, 3, 4, 5, 6];

この配列を受け取り、元の配列の先頭から該当位置までの全要素を足し合わせた値を対応する要素とする、新しい配列を返すJavaScript関数を作成する必要があります。

したがって、上記の配列の場合、期待される出力は次のようになります。

const output = [1, 3, 6, 10, 15, 21];

コード例

この処理を実現するコードは以下のとおりです。

const arr = [1, 2, 3, 4, 5, 6];
const findCumulativeSum = arr => {
    const creds = arr.reduce((acc, val) => {
        let { sum, res } = acc;
        sum += val;
        res.push(sum);
        return { sum, res };
    }, {
        sum: 0,
        res: []
    });
    return creds.res;
};
console.log(findCumulativeSum(arr));

コードの解説

このコードでは、Array.prototype.reduce() メソッドを活用しています。アキュムレーターとして { sum, res } というオブジェクトを初期化し、配列の各要素を先頭から順に処理していきます。

ループの各ステップでは、これまでの合計値 sum に現在の要素 val を加算して更新し、その値を結果配列 res の末尾に追加します。最後に res を返すことで、累積合計を格納した新しい配列が得られます。

なお、同じ処理は以下のようにシンプルに書くこともできます。

const arr = [1, 2, 3, 4, 5, 6];
let total = 0;
const output = arr.map(num => total += num);
console.log(output); // [ 1, 3, 6, 10, 15, 21 ]

実行結果

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

[ 1, 3, 6, 10, 15, 21 ]