JavaScriptで合計値が最大となるサブ配列のインデックスを返す方法

数値を要素に持つ「配列の配列」（ネストされた配列）が与えられたとき、各サブ配列の合計値を比較し、その合計が最大となるサブ配列のインデックスを返す関数を作成してみましょう。同じ最大値を持つサブ配列が複数存在する場合は、最初に出現したもののインデックスを返すのが条件です。

それでは、実際のコードを見ていきましょう。

コード例

const arr = [[4, 5, 1, 3], [13, 27, 18, 26], [32, 35, 37, 39], [1000, 1001, 857, 1]];
const findMaxSubArray = (arr) => {
   const add = (array) => array.reduce((acc, val) => acc + val);
   return arr.reduce((acc, val, ind) => {
      const sum = add(val);
      if(sum > acc.sum){
         return {
            index: ind,
            sum
         };
      };
      return acc;
   }, {
      index: -1,
      sum: -Infinity
   }).index;
};
console.log(findMaxSubArray(arr));

処理の流れ

まず、ヘルパー関数 add を定義し、reduce() メソッドを使って配列内の全要素の合計を計算できるようにしています。

次に、外側の配列に対しても reduce() を適用します。アキュムレータの初期値は { index: -1, sum: -Infinity } としており、これによりどのような負の値でも正しく比較できるようになっています。ループの中では各サブ配列の合計を順番に算出し、これまでの最大値よりも大きい場合のみ、インデックスと合計値でアキュムレータを更新します。

比較に厳密な不等号（>）を使用しているため、合計が同点のサブ配列が複数ある場合でも、後続のものは上書きされず、必ず最初のサブ配列のインデックスが返されます。

出力

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

3

この例では、4番目のサブ配列 [1000, 1001, 857, 1] の合計が 2859 となり最も大きいため、そのインデックスである 3 が出力されます。