Как переписать код в одну строку?

Elvis, Это было тестовое задание, которое я не смог решить. Вот и пытаюсь найти ответ.