Теперь вот алгоритм работает

И сходится с результатами 1.py