Не судите строго за топорное выполнение. Я только учусь.
Решение.
a = b = c = d = 1 # a - отвечает за цифры 1, 3, 7, 9, b - за 2, 8, c - за 4, 6, d - за 0
k = input()
for i in range(int(input())):
a, b, c, d = b + c, 2 * a, 2 * a + d, 2 * c
print(a if k in '1379' else b if k in '28' else c if k in '46' else d if k == '0' else 0)
Не судите строго за топорное выполнение. Я только учусь.
Решение.