3. Демоверсия ЕГЭ по информатике 2021

3.2. Решение itertools

         Игорь составляет таблицу кодовых слов для передачи сообщений, каждому сообщению соответствует своё кодовое слово. В качестве кодовых слов Игорь использует трехбуквенные слова, в которых могут быть только  буквы Ш, К, О, Л, А, причем буква К появляется ровно 1 раз. Каждая из других допустимых букв может встречаться в кодовом слове любое количество раз или не встречаться совсем. Сколько различных кодовых слов  может использовать Игорь?

#Подключаем библиотеку itertools
from itertools import *
#Создаем переменную со всеми комбинациями
p=product("ШКОЛА",repeat=3)
#Создаем переменную для подсчета кол-ва
#комбинаций в которых ровно одна К
res=0
#Перебираем все комбинации
for c in p:
    #Если в текущей комбинации одна К
    if(c.count("К")==1):
        #Увеличиваем на 1 кол-во ответов
        res+=1
#Печатаем на экран результат
print(res)

Ответ: 48

Источник: Демоверсия ЕГЭ по информатике 2021.