Pythonで文字列が別の文字列の接尾辞（サフィックス）かどうかを判定する方法

問題の概要

2つの文字列 s と t が与えられたとき、s が t の接尾辞（サフィックス）であるかどうかを判定します。

例えば、s = "ate"、t = "unfortunate" の場合、"unfortunate" という文字列は "ate" で終わっているため、出力は True になります。

解決手順

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

• s_len に s の長さ、t_len に t の長さを格納する
• s_len が t_len より大きい場合、s の方が長いため接尾辞になり得ず、False を返す
• i を 0 から s_len - 1 までループし、両方の文字列を末尾から順に1文字ずつ比較する
• 1文字でも一致しない場合は False を返す
• すべての文字が一致した場合は True を返す

実装例

以下のコードで、実際の動作を確認してみましょう。

def solve(s, t):
    s_len = len(s)
    t_len = len(t)
    if (s_len > t_len):
        return False
    for i in range(s_len):
        if(s[s_len - i - 1] != t[t_len - i - 1]):
            return False
    return True

s = "ate"
t = "unfortunate"
print(solve(s, t))

入力

"ate", "unfortunate"

出力

True

より簡単な方法：endswith() メソッド

Pythonでは、文字列型が標準で提供している endswith() メソッドを使うことで、同じ処理を1行で実現できます。

s = "ate"
t = "unfortunate"
print(t.endswith(s))  # True

endswith() は、対象の文字列が引数に指定した文字列で終わっているかどうかを判定し、その結果をブール値（True / False）で返します。実際の開発では、車輪の再発明を避けるため、この組み込みメソッドを使用するのが最もシンプルで推奨される方法です。

計算量について

どちらの方法でも、時間計算量は O(len(s))、空間計算量は O(1) となります。s の長さ分だけ比較を行うため、非常に効率的な処理です。