Интервьюер указал на ошибку: в коде используется max, хотя задача требует найти кратчайшее (минимальное) расстояние. Как исправить?