Python obi diz que minha solução esta errada

[lczx0129]

Estou tentando resolver um problema da Obi que se chama distanciamento entre amigos

eu fiz um programa para resolver o problema mas o sistema da obi diz que meu algoritmo

esta errado.eu não estou conseguindo achar o problema. poderia dizer o que esta errado

 

Link do Exercício : https://olimpiada.ic.unicamp.br/pratique/p1/2019/f1/amigos/

 

programa:

 

 

[Simon Viegas]

@lczx0129, por favor, poste o texto do código.