Задание 8. Информатика. ЕГЭ. Поляков-7413

(А. Минак) Все пятибуквенные слова, составленные из букв А, Э, Р, О, Б, У, С записаны в алфавитном порядке и пронумерованы. Вот начало списка:

1. ААААА
2. ААААБ
3. ААААО
4. ААААР
5. ААААС
6. ААААУ
7. ААААЭ
8. ...

Определите в этом списке количество слов с чётными номерами, которые содержат как минимум две буквы Р, между которыми стоит одна другая буква (например, РАР или РСР) и при этом не содержат в своей записи ни одной буквы У.

from itertools import product

def valid(s):
    if s.count('У'):
        return False
    if s.count('Р') < 2:
        return False
    elif any(r in s for r in ('РАР', 'РЭР', 'РОР', 'РБР', 'РСР')):
        return True
    return False

q, i = 0, 0

for p in product('АБОРСУЭ', repeat=5):
    i += 1
    q += int(i % 2 == 0 and valid(''.join(p)))

print(q)