Databricks interview question

Coding question about data structures and processing times.