Databricks interview question

Find K-th largest element in an array