Pergunta de entrevista da empresa Databricks

Simple coding question, no fancy algorithms.