Pythonで文字列内の連続する要素の最大積を求める方法

Pythonにはデータ操作のための強力なライブラリが数多く用意されています。そのような中、大きな文字列の一部となっている連続する2つの数字の積の最大値を求めたいという場面に出会うことがあります。本記事では、その問題を解決するための具体的な方法をいくつか紹介します。

zipとmaxを使う方法

まず文字列をリストに変換します。次に、スライスを利用して連続する要素同士のペアを作成します。各ペアに対して乗算を行い、その結果の中からmax関数で最大値を取得します。

例えば、zip(Astring, Astring[1:])は「5と2」「2と3」「3と8」「8と5」「5と2」「2と1」というペアを順に生成します。それぞれの積は10、6、24、40、10、2となるため、最大値の40が求める答えとなります。

サンプルコード

Astring = '5238521'
# 対象の文字列
print('対象の文字列 : ', Astring)
# 文字列をリストに変換
Astring = list(Astring)
print('リストに変換した文字列:\n', Astring)
# max()を使用して最大積を取得
res = max(int(a) * int(b) for a, b in zip(Astring, Astring[1:]))
# 結果を出力
print('連続する要素の最大積 : ', res)

実行結果

上記のコードを実行すると、以下のような結果が得られます。

対象の文字列 : 5238521
リストに変換した文字列:
['5', '2', '3', '8', '5', '2', '1']
連続する要素の最大積 : 40

mapとmaxを使う方法

こちらも基本的な考え方は上記と同じです。ただし、この方法ではmap関数を使って連続する整数のペアを継続的に生成します。そして、operatorモジュールのmul関数を使ってペア内の数値を乗算し、最後にmax関数を適用することで結果の最大値を取得します。mapは遅延評価されるため、大きなデータでもメモリ効率よく処理できる点が特徴です。

サンプルコード

from operator import mul
Astring = '5238521'
# 対象の文字列
print('対象の文字列 : ', Astring)
# 文字列をリストに変換
Astring = list(Astring)
print('リストに変換した文字列:\n', Astring)
# map()とmul()を組み合わせて最大積を取得
res = max(map(mul, map(int, Astring), map(int, Astring[1:])))
# 結果を出力
print('連続する要素の最大積 : ', res)

実行結果

上記のコードを実行すると、以下のような結果が得られます。

対象の文字列 : 5238521
リストに変換した文字列:
['5', '2', '3', '8', '5', '2', '1']
連続する要素の最大積 : 40

まとめ

どちらの方法も簡潔なコードで連続要素の最大積を求められます。zipを使った方法は可読性が高く、mapとoperator.mulを組み合わせた方法は関数型スタイルで処理を組み立てられるのが魅力です。コードの目的や好みに応じて使い分けるとよいでしょう。