Pythonで単語リストのスコアを計算するプログラムの書き方

配列の中にいくつかの単語が入っているとします。これらの単語はすべて小文字で構成されています。ここで、次のルールに基づいて単語セット全体の合計スコアを求めることを考えましょう。

• 母音は [a, e, i, o, u, y] の6文字とみなします。
• ある単語に含まれる母音の数が偶数の場合、その単語のスコアは 2 になります。
• それ以外（母音の数が奇数）の場合、その単語のスコアは 1 になります。
• 単語セット全体のスコアは、各単語のスコアをすべて足し合わせた値となります。

具体例

たとえば、入力が次のような単語リストだったとします。

words = ["programming", "science", "python", "website", "sky"]

この場合の出力は 6 になります。理由は以下の通りです。

• 「programming」：母音が3個（o, a, i）→ スコア 1
• 「science」：母音が3個（i, e, e）→ スコア 1
• 「python」：母音が2個（o, y）→ スコア 2
• 「website」：母音が3個（e, i, e）→ スコア 1
• 「sky」：母音が1個（y）→ スコア 1

したがって、1 + 1 + 2 + 1 + 1 = 6 という結果になります。

アルゴリズムの手順

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

1. 合計スコアを格納する変数 score を 0 で初期化します。
2. リスト内の各単語について、以下の処理を繰り返します。
   1. 母音のカウント用変数 num_vowels を 0 で初期化します。
   2. 単語内の各文字を順番に調べ、母音であれば num_vowels を 1 増やします。
   3. num_vowels が偶数なら score に 2 を加算し、奇数なら 1 を加算します。
3. 最終的な score を返します。

実装例

それでは、実際のPythonコードを見てみましょう。

def solve(words):
    score = 0
    for word in words:
        num_vowels = 0
        for letter in word:
            if letter in ['a', 'e', 'i', 'o', 'u', 'y']:
                num_vowels += 1
        if num_vowels % 2 == 0:
            score += 2
        else:
            score += 1
    return score

words = ["programming", "science", "python", "website", "sky"]
print(solve(words))

入力

["programming", "science", "python", "website", "sky"]

出力

6

補足：より簡潔な書き方

Pythonでは、ジェネレータ式を使うことで上記のコードをさらに簡潔に書くこともできます。

def solve(words):
    vowels = set('aeiouy')
    return sum(2 if sum(c in vowels for c in word) % 2 == 0 else 1 for word in words)

このように、母音を集合（set）として定義しておくと判定が高速になり、コードも読みやすくなります。計算量は単語の総文字数に比例する O(N) となり、非常に効率的です。