JavaScriptで文字列を結合して回文（パリンドローム）ペアを見つける方法

問題の概要

文字列の配列を引数として受け取るJavaScript関数を作成します。この関数は、配列内の文字列同士を結合したときに回文（パリンドローム）となるすべてのインデックスペアを、配列の配列として返す必要があります。

例えば、関数への入力が次のような場合を考えてみましょう。

const arr = ['tab', 'cat', 'bat'];

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

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

出力の解説

これは、「battab」と「tabbat」がどちらも回文であるためです。つまり、インデックス0の「tab」とインデックス2の「bat」は、どちらの順序で結合しても回文になります。

コード例

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

const arr = ['tab', 'cat', 'bat'];

// 文字列が回文かどうかを判定するヘルパー関数
const isPalindrome = (str = '') => {
    let i = 0;
    let j = str.length - 1;
    while (i < j) {
        if (str[i] != str[j]) return false;
        i++;
        j--;
    };
    return true;
};

// 回文になるペアのインデックスをすべて見つけるメイン関数
const palindromePairs = (arr = []) => {
    const res = [];
    for (let i = 0; i < arr.length; i++) {
        for (let j = i + 1; j < arr.length; j++) {
            if (isPalindrome(arr[i] + arr[j])) {
                res.push([i, j])
            }
            if (isPalindrome(arr[j] + arr[i])) {
                res.push([j, i])
            };
        };
    };
    return res;
};

console.log(palindromePairs(arr));

コードの解説

ここでは、まずisPalindrome()というヘルパー関数を定義しています。この関数は、文字列の先頭と末尾から順番に文字を比較していくことで、その文字列が回文かどうかを判定します。途中で一致しない文字が見つかれば即座にfalseを返し、最後まで比較できればtrueを返します。

続いてメインのpalindromePairs()関数では、二重ループを使って配列内のすべての文字列の組み合わせを生成しています。各ペアに対して「前＋後ろ」と「後ろ＋前」の両方の結合順序をチェックし、条件を満たすものについては、そのインデックスペアを結果配列resに追加していきます。なお、同じペアでも結合の順序によって回文になる・ならないが変わるため、両方向のチェックが必要です。

出力結果

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

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