JavaScriptで配列の要素間に演算子オブジェクトを挿入する方法

次のような数値の配列があるとします。

const numbers = [1, 6, 7, 8, 3, 98];

この数値の配列を、各要素が「value」というキーを持ち、その値として元の配列の要素を格納するオブジェクトの配列へと変換することを目指します。
それに加えて、隣り合う2つの要素の間には「operation」というキーを持つオブジェクトを挿入し、その値としては +、-、*、/ の中から順番に1つずつ使用します。

したがって、numbers 配列の場合、期待される出力は以下のようになります。

[
    { "value": 1 }, { "operation": "+" }, { "value": 6 }, { "operation": "-"},
    { "value": 7 }, { "operation": "*" }, { "value": 8 }, { "operation":"/" },
    { "value": 3 }, { "operation": "+" }, {"value": 98}
]

それでは、この変換を実現するコードを実際に書いてみましょう。

サンプルコード

const numbers = [1, 6, 7, 8, 3, 98, 3, 54, 32];
const insertOperation = (arr) => {
    const legend = '+-*/';
    return arr.reduce((acc, val, ind, array) => {
        acc.push({
            "value": val
        });
        if(ind < array.length-1){
            acc.push({
                "operation": legend[ind % 4]
            });
        };
        return acc;
    }, []);
};
console.log(insertOperation(numbers));

コードの解説

このコードでは、Array.prototype.reduce() メソッドを活用しています。reduce は第2引数として渡された空の配列 [] を初期値（アキュムレータ）とし、元の配列の各要素を先頭から順番に処理していきます。

各ステップでは、まず現在の要素を格納した { value: 要素 } オブジェクトを結果の配列に追加します。その後、現在の要素が配列の最後の要素でない場合（ind < array.length - 1）、続けて { operation: 演算子 } オブジェクトも追加します。これにより、値と演算子が交互に並ぶ構造が自然に実現されます。

演算子の選択には、文字列 '+-*/' を用意し、legend[ind % 4] の形式で参照しています。インデックスを4で割った余りを利用することで、+、-、*、/ の4種類の演算子が先頭から順に循環して使用される仕組みになっています。

出力

上記のコードをコンソールで実行すると、出力は次のようになります。

[
    { value: 1 }, { operation: '+' },
    { value: 6 }, { operation: '-' },
    { value: 7 }, { operation: '*' },
    { value: 8 }, { operation: '/' },
    { value: 3 }, { operation: '+' },
    { value: 98 }, { operation: '-' },
    { value: 3 }, { operation: '*' },
    { value: 54 }, { operation: '/' },
    { value: 32 }
]