Question d’entretien chez Meta

Top k, solved using quick select