会社名から最も多く出現する上位3文字を抽出するPythonプログラム

文字列として会社名が与えられたとします。この会社名の中から最も頻繁に出現する3文字を、以下のルールに従って抽出します。

• 出現回数が多い上位3文字を選ぶ
• 出現回数の降順で並べる
• 出現回数が同じ文字がある場合は、アルファベット順に並べる

例えば、入力が s = "TUTORIALSPOINT" の場合、出力は [[3, 'T'], [2, 'I'], [2, 'O']] となります。「T」が3回、「I」と「O」がそれぞれ2回出現しているためです。

解決の手順

この問題は、以下の手順で解くことができます。

• x := 文字列 s に含まれる各文字とその出現回数を対応付けたマップ(Counter)を作成する
• res := 結果を格納する新しいリストを作成する
• x 内の各文字 i について、ペア (出現回数 x[i], 文字 i) を res に追加する
• res をアルファベット順にソートする
• res を出現回数の降順で再度ソートする
• res の先頭3要素を返す

ここでポイントとなるのは、ソートを2段階で行う点です。Python の sorted() は安定ソートであるため、まずアルファベット順に並べた後に出現回数の降順でソートすると、出現回数が同じ文字は自動的にアルファベット順のまま維持されます。

実装例

以下のコードを見て、具体的な動作を確認しましょう。

from collections import Counter
def solve(s):
   x = Counter(s)
   res = []

   for i in x:
      res.append([x[i], i])

   res = sorted(res, key=lambda cnt: cnt[1])
   res = sorted(res, key=lambda cnt: cnt[0], reverse=True)

   return res[:3]

s = "TUTORIALSPOINT"
print(solve(s))

入力

"TUTORIALSPOINT"

出力

[[3, 'T'], [2, 'I'], [2, 'O']]