Pythonで2つの文字列を連結して特定の文字列を生成できるかどうかを判定する方法

この記事では、2つの文字列 s と t、およびもう1つの文字列 r が与えられたときに、r が s + t または t + s（連結）として表現できるかどうかを判定する方法を解説します。

例えば、s = "world"、t = "hello"、r = "helloworld" という入力の場合、"helloworld"（r）は "hello"（t）+ "world"（s）と一致するため、出力は True になります。

解決のアプローチ

この問題は、以下の手順で解くことができます。

• 長さのチェック： r の長さが s と t の長さの合計と異なる場合、連結では絶対に作れないため False を返します。
• パターン1の確認： r が s で始まり、かつ t で終わる場合、True を返します。
• パターン2の確認： r が t で始まり、かつ s で終わる場合、True を返します。
• 上記のいずれにも該当しない場合は False を返します。

ここで使用するのは、Pythonの組み込みメソッドである startswith()（指定した文字列で始まるかを判定）と endswith()（指定した文字列で終わるかを判定）です。これらを組み合わせることで、簡潔かつ効率的に判定できます。

実装コード例

def solve(s, t, r):
    if len(r) != len(s) + len(t):
        return False

    if r.startswith(s):
        if r.endswith(t):
            return True
            
    if r.startswith(t):
        if r.endswith(s):
            return True
    
    return False  

s = "world"
t = "hello"
r = "helloworld"
print(solve(s, t, r))

入力

"world", "hello", "helloworld"

出力

True

処理の流れのポイント

最初に長さを比較することで、明らかに条件を満たさないケースを早期に除外できます。これにより、不要な文字列操作を避けられます。また、長さの合計が一致している場合でも、先頭と末尾の両方が正しい順序で対応していない限り False となるため、このアルゴリズムは常に正確な結果を返します。