Подсчёт комбинаций: Определите количество пятизначных чисел; 8, 3, 5, 6 · ЕГЭ №8

from itertools import product
digits = '012345678'
res = []
for i in product(digits, repeat=5):
    if i[0] == '0': continue
    if i.count('3') != 1: continue
    pos3 = i.index('3')
    flag = True
    if pos3 > 0 and i[pos3-1] in '5678': flag = False
    if pos3 < 4 and i[pos3+1] in '5678': flag = False
    if flag: res.append(i)
print(len(res))