Pergunta de entrevista da empresa Citadel

Pretty normal algorithm coding question