Pythonで英数字文字列内の数字の合計を求めるプログラム

「0」から「9」までの数字と小文字の英字で構成される英数字の文字列 s があるとします。このとき、s に含まれる数値の合計を求める必要があります。ただし、数字が連続している場合は、それらを1つの数値として扱うものとします。

例えば、入力が s = "hello25world63power86" の場合、出力は 174 になります。これは 25 + 63 + 86 = 174 となるためです。

解決のアプローチ

この問題を解くために、以下の手順に従います。

• ret := 0、curr := 0 として初期化します

• 文字列 s 内の各文字 ch について、次の処理を行います

  • ch が数字の場合

    • curr := 10 * curr + (ch を整数に変換した値)

  • それ以外の場合(英字の場合)

    • ret := ret + curr

    • curr := 0 にリセット

• 最後に ret + curr を返します

このアルゴリズムのポイントは、変数 curr を使って連続する数字を桁ごとに組み立てていくことです。例えば「25」という2桁の数字は、まず「2」を読み込んで curr = 2 となり、次に「5」を読み込んだ時点で curr = 2 × 10 + 5 = 25 となります。英字が出現したタイミングで、それまでに組み立てた数値を合計 ret に加算し、curr をリセットします。

実装例

理解を深めるために、以下の実装例を見てみましょう。

from string import digits
def solve(s):
   ret = 0
   curr = 0
   for ch in s:
      if ch in digits:
         curr = 10 * curr + int(ch)
      else:
         ret += curr
         curr = 0
   return ret + curr

s = "hello25world63power86"
print(solve(s))

コードの解説

ここでは string モジュールの digits 定数を使用して、文字が数字かどうかを判定しています。digits は "0123456789" という文字列であり、ch in digits の条件式で簡単に数字判定が可能です。ループ終了後に ret + curr を返すのは、文字列の末尾が数字で終わっている場合に、まだ加算されていない curr の値を合計に含めるためです。

入力

"hello25world63power86"

出力

174