Casas de palitos de fósforo

Este desafio ajudará você a interpretar relações matemáticas tanto algebricamente quanto geometricamente.

Crie uma função que receba um número (step) como argumento e retorne o número de palitos de fósforo nesse passo. Veja os passos 1, 2 e 3 na imagem acima.

Exemplos

match_houses(1) ➞ 6

match_houses(4) ➞ 21

match_houses(87) ➞ 436

Observações

• O passo 0 retorna 0 palitos de fósforo.
• A entrada (step) sempre será um inteiro não negativo.
• Pense na entrada (step) como o número total de casas que foram conectadas.