правильный выбор главного элемента

TODO условия остановки
правильные условия остановки
убрал лишнее
доработка