Secuencia visual de bloques en 3D

Esta es una secuencia de bloques en tres dimensiones:

Crea una función que reciba un número de paso no negativo y devuelva la cantidad de bloques que hay en ese paso.

Ejemplos

blocks(1) ➞ 5

blocks(5) ➞ 39

blocks(2) ➞ 12

Notas

• El paso 0 devuelve 0.
• La entrada siempre es un número entero no negativo.
• Consulta la pestaña Recursos para ver un video sobre cómo encontrar secuencias cuadráticas.