Casas de palitos de fósforo

Este desafío te ayudará a interpretar relaciones matemáticas tanto algebraica como geométricamente.

Crea una función que reciba un número (step) como argumento y devuelva la cantidad de palitos de fósforo que hay en ese paso. Consulta los pasos 1, 2 y 3 en la imagen de arriba.

Ejemplos

match_houses(1) ➞ 6

match_houses(4) ➞ 21

match_houses(87) ➞ 436

Notas

• El paso 0 devuelve 0 palitos de fósforo.
• La entrada (step) siempre será un entero no negativo.
• Considera la entrada (step) como el número total de casas que se han conectado.