Sequência visual de blocos em 3D

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

Passos 1 a 5 da sequência

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.