【JavaScript】ネストされた配列の全要素数（ディープカウント）を求める方法

問題：ネストされた配列の全要素数を数える

入れ子（ネスト）構造を持つ配列を受け取り、その中に存在するすべての要素の総数（ディープカウント）を返すJavaScript関数を作成します。

入力

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

出力

const output = 7;

出力が7になるのは、第1階層に「1」「2」「[3, 4, [5]]」の3つ、第2階層に「3」「4」「[5]」の3つ、第3階層に「5」の1つの要素が存在し、入れ子の配列そのものも1つの要素としてカウントされるためです。すなわち 3 + 3 + 1 = 7 となります。

サンプルコード

const arr = [1, 2, [3, 4, [5]]];
const deepCount = (arr = []) => {
    return arr
    .reduce((acc, val) => {
        return acc + (Array.isArray(val) ? deepCount(val) : 0);
    }, arr.length);
};
console.log(deepCount(arr));

コードの解説

Array.prototype.reduce() メソッドを使って配列を順番に走査し、各要素が配列かどうかを Array.isArray() で判定しています。走査の途中で入れ子の配列が見つかった場合は、同じ関数を再帰的に呼び出し、その内側に含まれる要素数も合計に加算していきます。

この実装のポイントは、reduce の初期値として arr.length を渡している点です。まず現在の階層の要素数がベースとしてカウントされ、そこへ再帰処理によって得られたネスト内の要素数が積み上がっていくため、どれだけ深く入れ子になっていても正確な総数を取得できます。

実行結果

7