JavaScriptで配列内の1を先頭へ、0を末尾へ分離する方法

JavaScriptでは、0・1・その他の数値が混在する配列に対して、「すべての1を配列の先頭に集め、すべての0を末尾に集めたい」という要件が発生することがあります。本記事では、このような配列の分離（セグリゲート）処理を実装する方法を解説します。

実装のポイント

今回のアプローチでは、以下の手順で処理を行います。

• slice() で元の配列のコピーを作成し、元データを変更しないようにする
• 配列を走査し、要素が 0 なら splice() で取り除いた後、push() で末尾へ移動する
• 要素が 1 なら splice() で取り除いた後、unshift() で先頭へ移動する
• それ以外の数値は元の位置にそのまま残す

コード例

const arr = [3, 2, 1, 8, 9, 0, 1, 9, 0, 2, 1, 0, 2, 0, 1, 0, 1, 1, 4, 0, 3];
const segregate = arr => {
    const copy = arr.slice();
    for(let i = 0; i < copy.length; i++){
        if(copy[i] === 0){
            copy.push(copy.splice(i, 1)[0]);
        }else if(copy[i] === 1){
            copy.unshift(copy.splice(i, 1)[0]);
        }
    }
    return copy;
};
console.log(segregate(arr));

出力結果

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

[
    1, 1, 1, 3, 2, 8, 9,
    1, 9, 2, 2, 1, 1, 4,
    3, 0, 0, 0, 0, 0, 0
]

コードの解説

ループ内で splice() を呼び出すたびに配列の長さと各要素のインデックスが変化することに注意してください。特に0を末尾へ移動した直後は後続の要素が1つ前に詰まるため、直後の要素が走査対象から外れる（スキップされる）ことがあります。そのため、上記の出力例のように一部の1が配列の中間に残る場合があります。

より堅牢な実装：filter()を活用する方法

インデックス操作による副作用を避けたい場合は、filter() で要素を振り分けてから連結する方法がシンプルで確実です。

const segregate = arr => {
    const ones = arr.filter(n => n === 1);
    const zeros = arr.filter(n => n === 0);
    const others = arr.filter(n => n !== 0 && n !== 1);
    return [...ones, ...others, ...zeros];
};

この実装では、すべての1が必ず先頭に、すべての0が必ず末尾に配置され、その他の数値は元の順序を保ったまま中間に並びます。計算量もO(n)で安定しており、可読性・保守性の面でも有利です。用途に応じて、両者の実装を使い分けるとよいでしょう。