Pergunta de entrevista da empresa Databricks

Leetcode style question, using heap