Pergunta de entrevista da empresa Arista Networks

Two sum variation in python