Можно ли улучшить код решения задачи с учетом сортировки?