Задание 5. Информатика. ЕГЭ. Поляков-5721

(А. Игнатюк) Исполнитель «Аполлон» получает на вход четырёхзначное число \(N\) и строит новое число \(R\) по следующим правилам:

1. Если число \(N\) начинается с чётной цифры, то число \(R\) вычисляется как сумма первой и третьей цифр и модуля разности второй и четвёртой цифр.
2. Если число \(N\) начинается с нечётной цифры, то цифры числа \(N\) располагают в неубывающем порядке. Число \(R\) вычисляется как сумма цифр в двоичной записи полученного числа.

Сколько существует чисел \(N,\) для которых результат работы алгоритма будет более \(20?\)

def R(N):
    sn = str(N)
    if int(sn[0]) % 2 == 0:
        ans = int(sn[0]) + int(sn[2]) + abs(int(sn[1]) - int(sn[3]))
    else:
        tmp = int(''.join(sorted(list(sn))))
        ans = bin(tmp).count('1')
    return ans

print(len([N for N in range(1000, 10_000) if R(N) > 20]))