Pythonで2つの文字列を交互に結合する方法(文字列インターリーブ)

文字列インターリーブとは

2つの文字列 s と t が与えられたとき、最初の文字列 s の文字から始めて、両者の文字を交互に並べた新しい文字列を作ることを「インターリーブ(交互結合)」と呼びます。片方の文字列が先に尽きた場合は、余った文字をそのまま末尾に追加します。

例えば、入力が s = "abcd"、t = "pqrstu" の場合、出力は "apbqcrdstu" となります。

アルゴリズムの手順

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

• 結果を格納する変数 res を空文字列として初期化します。
• インデックス用の変数 i を 0 に設定します。
• m を s と t の長さのうち小さい方の値とします。
• i < m の間、次の処理を繰り返します。
  • res に s[i] と t[i] をこの順で連結して追加します。
  • i を 1 増やします。
• ループ終了後、res に s のインデックス i 以降の部分と t のインデックス i 以降の部分を連結して返します。

実装例(Pythonコード)

class Solution:
    def solve(self, s, t):
        res = ""
        i = 0
        m = min(len(s), len(t))
        while i < (m):
            res += s[i] + t[i]
            i += 1
        return res + s[i:] + t[i:]

ob = Solution()
s = "abcd"
t = "pqrstu"
print(ob.solve(s, t))

入力

"abcd","pqrstu"

出力

apbqcrdstu

コードの解説

まず、短い方の文字列の長さ(m)だけループを回し、各位置の文字を s → t の順で交互に res へ追加していきます。この例では s(4文字)の方が t(6文字)より短いため、4回のループで "apbqcrds" が作られます。

続いて、スライス s[i:] と t[i:] を使って余った文字を取り出し、末尾に連結します。s には余りがなく、t の残り "tu" が追加されるため、最終的な出力は "apbqcrdstu" となります。