Sequência visual de blocos em 3D

Aqui está 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.