Question d’entretien chez Databricks

Leetcode style question, using heap