Pregunta de entrevista de Arista Networks

Two sum variation in python