JavaScriptで配列内の部分文字列とスーパー文字列の組み合わせをすべて検索する方法

本記事では、文字列の配列を引数として受け取り、その配列内に存在するすべての「部分文字列」と「スーパー文字列（より長い側の文字列）」の組み合わせを検出し、該当する要素だけをまとめた配列を返すJavaScript関数の作成方法を解説します。

問題の概要

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

const arr = ["abc", "abcd", "abcde", "xyz"];

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

const output = ["abc", "abcd", "abcde"];

これは、「abc」と「abcd」がいずれも「abcde」の部分文字列となっているためです。一方、「xyz」は他のどの文字列とも部分文字列の関係にないため、結果には含まれません。

実装コード

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

const arr = ["abc", "abcd", "abcde", "xyz"];
const findStringCombinations = (arr = []) => {
    let i, j, res = {};
    for (i = 0; i < arr.length - 1; i++) {
        if (res[arr[i]]) {
            continue;
        };
        for (j = i + 1; j < arr.length; j++) {
            if (res[arr[j]]) {
                continue;
            }
            if (arr[i].indexOf(arr[j]) !== -1 || arr[j].indexOf(arr[i]) !== -1) {
                res[arr[i]] = true;
                res[arr[j]] = true;
            }
        };
    };
    const result = arr.filter(el => res[el]);
    return result;
};
console.log(findStringCombinations(arr));

出力結果

コンソールに出力される結果は次のとおりです。

[ 'abc', 'abcd', 'abcde' ]

コードの仕組み

この関数の処理の流れは以下のとおりです。

まず、二重ループによって配列内のすべての文字列ペアを比較します。String.prototype.indexOf() メソッドを使用し、一方の文字列がもう一方の中に含まれているかどうかを判定します。indexOf() の戻り値が -1 以外であれば、そのペアは部分文字列の関係にあるとみなされ、補助用オブジェクト res にフラグとして記録されます。

また、すでにフラグが立った文字列は continue でスキップされるため、同じ文字列が何度も評価される無駄がありません。

最後に Array.prototype.filter() を使って、フラグが立った要素のみを元の配列の順序のまま抽出し、結果として返しています。