Pythonで2つの文字列を交互にマージするプログラムの書き方

同じ長さを持つ2つの文字列 s と t が与えられたとします。このとき、両者の文字を交互に取り出して連結し、新しい1つの文字列を作ることを考えます。具体的には、s[i] に t[i] を連結し、その後インデックスを1つ進めて同じ操作を繰り返していくイメージです。

たとえば、入力が s = "hello"、t = "world" の場合、出力は "hweolrllod" となります。先頭から順に h・w・e・o・l・r・l・l・o・d と交互に並んでいることが確認できますね。

解決のための手順

この問題を解くには、以下のステップに従います。

• ステップ1: Python組み込みの zip() 関数を使い、s と t を対応する位置ごとにペア (s[i], t[i]) としてまとめます。
• ステップ2: 各ペアに対して s[i] + t[i] のように文字列を連結し、新しいリストを作成します。
• ステップ3: join() メソッドでリスト内のすべての要素を1つの文字列に結合し、結果として返します。

実装例

それでは、実際のコードを見てみましょう。

def solve(s, t):
    zipped = list(zip(s, t))
    zipped = map(lambda x: x[0]+x[1], zipped)
    return ''.join(zipped)

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

入力

"hello", "world"

出力

hweolrllod

コードの解説

まず list(zip(s, t)) によって、[('h', 'w'), ('e', 'o'), ('l', 'r'), ('l', 'l'), ('o', 'd')] というペアのリストが生成されます。次に map() とラムダ式 lambda x: x[0]+x[1] を組み合わせることで、各タプルの第1要素と第2要素を連結し、'hw'、'eo'、'lr'... という文字列のシーケンスを作ります。最後に ''.join() でこれらをすべてつなげれば、目的の交互マージ文字列が完成します。

なお、処理は文字列の長さ n に対して O(n) の時間計算量で動作するため、非常に効率的です。また、itertools.chain.from_iterable(zip(s, t)) を使うなど、より簡潔な別の実装方法も存在しますので、興味があれば試してみてください。