Databricks interview question

NDA, pretty standard coding questions