【JavaScript】文字列の一意な文字を配列にマッピングする方法

JavaScriptで、文字列を受け取り、その各文字を0から始まる連番にマッピングする関数を作成してみましょう。ルールはシンプルです。初めて登場した文字（重複していない文字）に出会うたびにカウントを1つ増やし、すでに登場したことのある文字には同じ番号を割り当てます。

問題の例

たとえば、次のような文字列が与えられたとします。

const str = 'heeeyyyy';

この場合、期待される出力は以下のとおりです。

const output = [0, 1, 1, 1, 2, 2, 2, 2];

結果を見ると、「h」には0、「e」には1、「y」には2が割り当てられており、同じ文字が続く間は同じ番号が使われていることがわかります。

コード例

以下が実際のコードです。

const str = 'heeeyyyy';
const mapString = str => {
    const res = [];
    let curr = '', count = -1;
    for(let i = 0; i < str.length; i++){
       if(str[i] === curr){
          res.push(count);
       }else{
          count++;
          res.push(count);
          curr = str[i];
       };
    };
    return res;
};
console.log(mapString(str));

コードの仕組み

この関数では、次の3つの要素を使ってマッピングを行っています。

• res：結果となる番号を格納する配列です。
• curr：直前に処理した文字を記憶しておく変数です。初期値は空文字列です。
• count：現在のマッピング番号を表します。初期値は -1 で、新しい文字が出るたびにインクリメントされるため、最初の文字には必ず 0 が割り当てられます。

ループの中では、現在の文字 str[i] が curr と一致していれば、そのまま現在の count を配列に追加します。一致しない場合（＝新しい文字が現れた場合）は、count を1増やしてから配列に追加し、curr を現在の文字で更新します。これにより、文字の切り替わりごとに番号が1つずつ増えていく仕組みです。

実行結果

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

[
    0, 1, 1, 1,
    2, 2, 2, 2
]