Casas de palitos de fósforo

Este desafio ajudará você a interpretar relações matemáticas de forma algébrica e geométrica.

Crie uma função que receba um número ($step) como argumento e retorne a quantidade de palitos de fósforo nessa etapa. Veja as etapas 1, 2 e 3 na imagem acima.

Exemplos

match_houses(1); // 6

match_houses(4); // 21

match_houses(87); // 436

Observações

• Na etapa 0, são usados 0 palitos de fósforo.
• A entrada ($step) será sempre um número inteiro não negativo.
• Considere que a entrada ($step) representa o número total de casas conectadas entre si.