Подсчёт комбинаций: Сколько существует десятичных четырёхзначных чисел, в которых · ЕГЭ №8

from itertools import product
digits = '0123456789'
res = []
for i in product(digits, repeat=4):
    if i[0] == '0': continue
    if len(set(i)) != 4: continue
    flag = True
    for j in range(3):
        if int(i[j]) % 2 == int(i[j+1]) % 2:
            flag = False
            break
    if flag: res.append(i)
print(len(res))