Домики из спичек

Это испытание поможет вам интерпретировать математические отношения как алгебраические, так и геометрические.

Создайте функцию, которая принимает число (шаг) в качестве аргумента и возвращает количество спичек на этом шаге. См. шаги 1, 2 и 3 на изображении выше.

Примеры

match_houses(1) ➞ 6

match_houses(4) ➞ 21

match_houses(87) ➞ 436

Примечание

• Шаг 0 возвращает 0 спичек.
• Входные данные (шаг) всегда будут неотрицательным целым числом.
• Подумайте о входных данных (шаг) как о общем количестве домов, которые были соединены вместе.