Domanda di colloquio di Databricks

Leetcode style question, using heap