написать короче

Всем привет! Написал тут решение данной задачи, но мне кажется что это какой-то костыльЕсли есть идеи, накидайте, пожалуйста