@g3nv3

Как сократить код?

Решаю простые задачи на комбинаторику, как можно сократить этот код?
from itertools import product

j = 0
for i in product("AB123", repeat = 8):
    k = ''.join(i)
    if(k.count("A")==2 and k.count("B")==0):
        j += 1
    elif(k.count("A")==1 and k.count("B")==1):
        j += 1
    elif(k.count("A")==0 and k.count("B")==2):
        j += 1
print(j)

Он работает, но мне кажется, что можно сделать как-то короче.
В задаче нужно узнать количество комбинаций, в которых будет ровно две буквы.
  • Вопрос задан
  • 100 просмотров
Пригласить эксперта
Ответы на вопрос 1
Vindicar
@Vindicar
RTFM!
if (k.count("A") + k.count("B")) == 2:
Ответ написан
Комментировать
Ваш ответ на вопрос

Войдите, чтобы написать ответ

Похожие вопросы