Sequência visual de blocos em 3D

Esta é uma sequência de blocos em três dimensões:

Crie uma função que receba um número de etapa não negativo e retorne a quantidade de blocos dessa etapa.

Exemplos

blocks(1) ➞ 5

blocks(5) ➞ 39

blocks(2) ➞ 12

Observações

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