Лида составляет слова из букв К, Р, Ы, Ш, А. Каждая гласная буква встречается в слове не более двух раз. Каждая согласная может стоять в слове на первой позиции, либо не встречаться вовсе. Сколько слов длиною более двух символов может составить Лида? НА ПИТОНЕ
СОСТАВИТЬ ПРОГРАММУ

from itertools import *

m = 0

p = 0

for i in range(2,5):

n = 0

word = product('ИА',repeat=i) ## i -это кол-во символов в слове

for letter in word:

if letter.count('И') <3 and letter.count('А') <3:

n += 1 ##кол-во слов, где каждая гласная встречается не более двух раз

if i !=2:

p +=n ## кол-во слов длиной более двух символов без согласных

m+=n*2 ## кол-во слов длиной более двух символов с 2мя согласными вначале слова

print(m+p)