JavaScriptで解くバナナ配布問題：配列を使った実装方法を解説

問題

n人が一列に並んでいるとします。この人たちに対して、次のルールに従ってバナナを配布していきます。

• まず、1人目に1本、2人目に2本…というように順番に渡し、n人目にn本を渡します。

• 次に、列の先頭に戻り、1人目にn+1本、2人目にn+2本…というように渡し、n人目に2n本を渡します。

• このプロセスを、バナナがなくなるまで繰り返します（毎回渡す本数を1本ずつ増やし、列の末尾に達したら先頭に戻ります）。途中でバナナが足りなくなった場合は、残りの本数をすべてその時点の相手に渡します。

ここで求められているのは、人数 num を第1引数、バナナの総本数 m を第2引数として受け取るJavaScript関数を作成することです。

この関数は、最終的なバナナの配布結果を表す、長さ num の配列を返す必要があります。

例えば、関数への入力が以下だったとします。

const num = 3;
const m = 10;

このとき、期待される出力は次の通りです。

const output = [5, 2, 3];

出力の解説

処理の流れを順に追ってみましょう。

1回目のターンでは res[0] += 1 が実行され、配列は [1, 0, 0] になります。

2回目のターンでは res[1] += 2 が実行され、配列は [1, 2, 0] になります。

3回目のターンでは res[2] += 3 が実行され、配列は [1, 2, 3] になります。

4回目のターンでは res[0] += 4 が実行されますが、この時点で残りのバナナは4本しかないため、すべて1人目に渡されます。最終的な配列は [5, 2, 3] となります。

コード例

この問題を解くコードは以下の通りです。

const num = 3;
const m = 10;
const distributeBananas = (num = 1, m = 1) => {
    const res = new Array(num).fill(0);
    let curr = 1;
    while(true){
        for(let i = 0; i < num; i++){
            if(m < curr){
                res[i] += m
                return res
            };
            res[i] += curr;
            m -= curr;
            curr++;
        };
    };
};
console.log(distributeBananas(num, m));

このコードでは、まず全員の受け取った本数を0で初期化した配列 res を用意し、次に渡す本数を表す変数 curr を1で初期化しています。無限ループの中で列を一周ずつ回り、各人に curr 本ずつ配っていきます。もし残りのバナナ数 m が curr より少なくなった時点で、残りをすべてその人に渡して結果を返します。

出力

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

[5, 2, 3]