Sequência visual de blocos em 3D

Uma sequência de blocos em três dimensões. Podemos escrever uma fórmula para ela:

Crie uma função que receba um número (step) como argumento e retorne a quantidade de blocos nessa etapa.

Exemplos

blocks(1) ➞ 5

blocks(5) ➞ 39

blocks(2) ➞ 12

Observações

• A etapa 0 obviamente deve retornar 0.
• A entrada é sempre um número inteiro não negativo.
• Consulte a aba Resources para ver um vídeo sobre como encontrar sequências quadráticas.