Choose fastest transportation mode on city grid | Databricks Coding Question