Sonder interview question

Variant of top K elements