про Козла

Есть такая задача, условие нижеНаписал свой код - решение в 3 из 10 тестов неверное, формула вроде правильная, на тестовом условии тоже ответ верный