JavaScriptでネストされた配列の加重和を計算する方法

問題

任意の深さまでネストされた配列 arr を唯一の引数として受け取り、その加重和（weighted sum）を計算して返す JavaScript 関数を作成する必要があります。

加重和とは、各要素に「その要素が属するネストの深さ（レベル）」を掛けた値を、配列全体にわたって合計したものです。

入力例と出力例

たとえば、関数に次の配列を渡した場合を考えてみましょう。

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

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

const output = 46;

なぜ 46 になるのか？

最上位の要素はレベル1、その中に入っている要素はレベル2、さらにその内側はレベル3 として扱います。各要素に対応するレベルを掛けて合計すると、次のように計算されます。

(4 * 1) + (7 * 1) + (6 * 2) + (1 * 2) + (5 * 3) + (2 * 3) = 46

実装コード

この問題は再帰呼び出しを使うことで、シンプルかつ読みやすく解決できます。

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

const findWeightedSum = (arr = [], level = 1, res = 0) => {
  for (let i = 0; i < arr.length; i++) {
    if (typeof arr[i] === 'number') {
      // 数値なら現在のレベルを掛けて加算
      res += (level * arr[i]);
    } else if (Array.isArray(arr[i])) {
      // 配列ならレベルを1つ深くして再帰的に処理
      res = findWeightedSum(arr[i], level + 1, res);
    }
  }
  return res;
};

console.log(findWeightedSum(arr));

実行結果

46

コードのポイント

• デフォルトパラメータ level = 1 と res = 0 を使うことで、最初の呼び出し時に特別な初期化を行わずに済みます。
• 要素が数値（typeof arr[i] === 'number'）であれば、そのレベルを掛けて累積変数 res に加算します。
• 要素が配列（Array.isArray(arr[i])）であれば、レベルを +1 して自分自身を再帰的に呼び出します。
• 重要なのは、return findWeightedSum(...) ではなく res = findWeightedSum(...) としている点です。return してしまうとループがそこで終了し、同じ階層に残った後続の要素が処理されなくなります。結果を res に代入することで、すべての要素を正しく走査できます。