JavaScriptで隣接要素の差に基づいてソート済み配列をグループ化する方法

以下のように、昇順に並べ替えられた整数の配列があると仮定します。

const arr = [ 1, 2, 3, 5, 6, 7, 10, 12, 17, 18];

このような配列を受け取るJavaScript関数を作成する必要があります。この関数は、次の条件を満たすように配列をグループ化しなければなりません。

• 同じグループ内の要素どうしの差は1以下であること

• 有効なグループとみなされるためには、各グループが2つ以上の要素を含んでいること

上記の条件に基づくと、期待される出力は次のようになります。

const output = [
    [1, 2, 3],
    [5, 6, 7],
    [17, 18]
];

実装例

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

const arr = [ 1, 2, 3, 5, 6, 7, 10, 12, 17, 18];
const groupNear = (arr = []) => {
    const res = [];
    for (let ind = 0; ind < arr.length; ind++) {
        let value = arr[ind];
        if (arr[ind - 1] + 1 === value) {
            res[res.length - 1].push(value);
        } else if (value + 1 === arr[ind + 1]) {
            res.push([value]);
        };
    };
    return res;
};
console.log(groupNear(arr));

出力結果

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

[ [ 1, 2, 3 ], [ 5, 6, 7 ], [ 17, 18 ] ]

コードの解説

この関数では、forループを使って配列の各要素を先頭から順に走査しています。注目しているのは「前の要素」と「次の要素」それぞれとの関係です。

まず、現在の要素が直前の要素と連続している（直前の要素＋1と一致する）場合、その要素は既存の連続グループの続きとみなせるため、結果配列の最後のグループに追加されます。

一方、直前の要素とは連続していないものの、次の要素と連続している場合（現在の要素＋1が次の要素と一致する場合）は、そこが新しい連続グループの始まりとなるため、新しいグループを作成してその値を格納します。

この判定をすべての要素に対して繰り返すことで、「隣接する要素どうしの差が1である要素」だけで構成されたグループのみが残り、単独の要素（たとえば例の10や12など、前後どちらとも連続しない要素）は自動的に除外されます。計算量はO(n)と配列の長さに比例して済むため、大きなデータセットでも効率的に動作します。