Domanda di colloquio di Citadel

Pretty normal algorithm coding question