Optimize a SQL query plan tree | Databricks Coding Question