Pregunta de entrevista de Databricks

Simple coding question, no fancy algorithms.