Pythonで電話番号を整形（リフォーマット）するプログラムの実装方法

ここでは、数字・スペース・ハイフン（-）で構成された電話番号の文字列を受け取り、決められたルールに従って読みやすい形式へ整形するプログラムをPythonで実装します。

整形のルール

• 文字列に含まれるすべてのスペースとハイフンを除去する
• 残った数字を左から順に3桁ずつのブロックにグループ化し、残りが4桁以下になったら終了する
• 最後に残った数字は、その桁数に応じて次のようにグループ化する
  • 2桁の場合：長さ2のブロック1つ
  • 3桁の場合：長さ3のブロック1つ
  • 4桁の場合：長さ2のブロック2つ（例：12-34）

こうして作られた各ブロックはハイフンで連結され、最終的な電話番号となります。

入出力の例

たとえば、入力が s = "9-6-84102-4 7-8" の場合、出力は 968-410-24-78 になります。

アルゴリズムの考え方

1. 結果を格納する digits と、現在処理中のブロックを保持する blk を空文字列として初期化します。
2. 入力文字列 s の各文字 i について以下を繰り返します。
   • i が数字なら blk に連結する
   • blk の長さが3になったら、blk とハイフンを digits に追加し、blk を空に戻す
3. ループ終了後、blk に残っている桁数で分岐します。
   • 0桁の場合：末尾の余分なハイフンを除いた digits をそのまま返す
   • 1桁の場合：直前の3桁ブロックから1桁を後ろに移動させ、「2桁＋2桁」の形に組み替えて返す（4桁を2-2に分割するルールを満たすため）
   • 2桁の場合：digits に blk をそのまま連結して返す

Pythonでの実装例

以下が実際のコードです。

def solve(s):
    digits = ""
    blk = ""
    for i in s:
        if i.isnumeric():
            blk += i
        if len(blk) == 3:
            digits += blk+"-"
            blk = ""

    if len(blk) == 0:
        return digits[:-1]
    elif len(blk) == 1:
        return digits[:-2]+"-"+digits[-2]+blk
    elif len(blk) == 2:
        return digits+blk

s = "9-6-84102-4 7-8"
print(solve(s))

入力

"9-6-84102-4 7-8"

出力

968-410-24-78

ポイントの解説

このアルゴリズムの肝は、残り1桁の場合の処理です。単純に3桁区切りで進めると、最後に1桁だけ余ってしまうことがあります。そこで、直前に作った3桁ブロックの末尾1桁を後ろに移動し、「2桁＋2桁」の2ブロックに再構成することで、4桁の分割ルールに自然に対応できます。計算量は文字列の長さをNとするとO(N)で、非常に効率的な実装となっています。