Pregunta de entrevista de Databricks

Leetcode style question, using heap