Pythonでソートされた母音文字列をカウントするプログラム

数nがあるとすると、母音（a、e、i、o、u）のみで構成されるサイズnの文字列の数を見つける必要があり、辞書式順序で並べ替えられます。文字列sは、すべての有効なインデックスiについて、s[i]がアルファベットのs[i + 1]と同じか、その前にある場合に辞書式順序で並べ替えられていると言えます。

したがって、入力がn =2の場合、["aa"、 "ae"、 "ai"、 "ao"、 "au"、 "ee"、 "eiなどの文字列が多数あるため、出力は15になります。 "、" eo "、" eu "、" ii "、" io "、" iu "、" oo "、" ou "、"uu"]。

これを解決するには、次の手順に従います-

• nが1と同じ場合、
  • 5を返す
• count：=サイズ6の配列で、最初は1で埋められています
• 3からnの範囲のiについては、
  • count [1]：=count [1] + count [2] + count [3] + count [4] + count [5]
  • count [2]：=count [2] + count [3] + count [4] + count [5]
  • count [3]：=count [3] + count [4] + count [5]
  • count [4]：=count [4] + count [5]
• 合計：=0
• 1から5の範囲のiについては、
  • 合計：=合計+ i * count [i]
• 合計を返す

例

理解を深めるために、次の実装を見てみましょう-

def solve(n):
   if n==1:
      return 5
   count = [1 for i in range(6)]
   for i in range(3,n+1):
      count[1] = count[1]+count[2]+count[3]+count[4]+count[5]
      count[2] = count[2]+count[3]+count[4]+count[5]
      count[3] = count[3]+count[4]+count[5]
      count[4] = count[4]+count[5]
   total = 0
   for i in range(1,6):
      total += i*count[i]
   return total

n = 2
print(solve(n))

入力

2

出力

15