Лида составляет слова из букв Л, И, Д, А. Каждая гласная буква встречается в слове не более двух раз. Каждая согласная может стоять в слове на первой позиции, либо не встречаться вовсе. Как узнать сколько слов длиною более двух символов может составить Лида?

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)

надеюсь питон знаешь,мб можно и короче написать