JavaScriptで配列をソートし、重複する数値ごとにサブ配列へグループ化する方法

JavaScriptでは、数値の配列を唯一の引数として受け取る関数を作成することが求められるケースがあります。この入力配列には、同一の値（重複した要素）が含まれている可能性があります。

ここで作成する関数は、配列を昇順に並べ替えたうえで、すべての同一の数値をそれぞれ独立したサブ配列にグループ分けして返す必要があります。

例

たとえば、入力配列が次のような場合 −

const arr = [5, 7, 5, 7, 8, 9, 1, 1];

期待される出力は以下のとおりです −

const output = [
    [1, 1],
    [5, 5],
    [7, 7],
    [8],
    [9]
];

実装コード

この処理を実現するコードは次のようになります −

const arr = [5, 7, 5, 7, 8, 9, 1, 1];
const sortAndGroup = (arr = []) => {
    let result = [];
    let groupArray;
    arr.sort((a, b) => a - b);
    for (let i = 0; i < arr.length; i++) {
        if (arr[i - 1] !== arr[i]) {
            groupArray = [];
            result.push(groupArray);
        };
        groupArray.push(arr[i]);
    };
    return result;
};
console.log(sortAndGroup(arr));

出力結果

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

[ [ 1, 1 ], [ 5, 5 ], [ 7, 7 ], [ 8 ], [ 9 ] ]

コードの解説

このアルゴリズムの仕組みはシンプルです。まず sort() メソッドと比較関数 (a, b) => a - b を使って、配列を数値として正しく昇順に並べ替えます。

続いてループ処理の中で、現在の要素 arr[i] と直前の要素 arr[i - 1] を比較します。値が異なる場合は新しい空のサブ配列を作成し、結果配列に追加します。値が同じであれば、現在開いているサブ配列にそのまま要素を追加していきます。

この方法なら、ソート部分が O(n log n)、グループ化部分が O(n) となり、全体として効率的に処理できます。また、アロー関数とデフォルト引数を使用することで、簡潔で安全な実装になっています。