Pregunta de entrevista de Anyscale (CA)

Q: How to implement this function in Python