Conteo simple

Eleva al cuadrado cada número entero desde 0 hasta n. Luego cuenta cuántas veces aparece el dígito d en las representaciones decimales de todos esos cuadrados.

Ejemplos

count_digits(10, 1) ➞ 4
# Matriz de cuadrados desde 0 hasta 10 = [0, 1, 4, 9, 16, 25, 36, 49, 64, 81, 100]
# El dígito 1 apareció 4 veces en la matriz

count_digits(25, 2) ➞ 9

count_digits(10, 1) ➞ 4

Notas

d siempre será 0<=d<10.