Pythonで2つの文字列を交互にマージする方法を解説

2つの文字列 s と t があるとします。これらを、s の文字から始めて交互に1文字ずつ取り出しながら結合（マージ）することを考えます。もし2つの文字列の長さが異なる場合は、余った文字をそのままマージ後の文字列の末尾に追加します。

例えば、入力が s = "major"、t = "general" の場合、出力は mgaejnoerral になります。t の方が長いため、余りの部分である ral が末尾に追加されるためです。

解決の手順

この問題は、以下のステップで解決できます。

• インデックス i と j をそれぞれ 0 で初期化する

• 結果を格納する空の文字列 result を用意する

• i が s の長さ未満かつ j が t の長さ未満の間、以下を繰り返す

  • result に s[i] と t[j] を順に連結する

  • i と j をそれぞれ1つ進める

• i が s の長さ未満の間、result に s[i] を連結し、i を進める

• j が t の長さ未満の間、result に t[j] を連結し、j を進める

• result を返す

それでは、実際の実装例を見てみましょう。

実装例

def solve(s, t):
    i = j = 0
    result = ""
    while i < len(s) and j < len(t):
        result += s[i] + t[j]
        i += 1
        j += 1
    while i < len(s):
        result += s[i]
        i += 1
    while j < len(t):
        result += t[j]
        j += 1
    return result

s = "major"
t = "general"
print(solve(s, t))

入力

"major", "general"

出力

mgaejnoerral

より簡潔な書き方：itertools.zip_longest を活用

Pythonの標準ライブラリ itertools.zip_longest を使うと、同じ処理をより短くエレガントに記述できます。この関数は、長い方の文字列に合わせて要素を組み合わせ、足りない部分を指定した値（デフォルトでは空文字）で埋めてくれます。

from itertools import zip_longest

def solve(s, t):
    return "".join(a + b for a, b in zip_longest(s, t, fillvalue=""))

s = "major"
t = "general"
print(solve(s, t))  # mgaejnoerral

どちらの方法でも計算量は O(n + m)（n、m は各文字列の長さ）となり、効率的に動作します。ループの挙動を理解したい場合は前者の実装、コードの簡潔さを重視する場合は zip_longest を使った後者の実装がおすすめです。