Pergunta de entrevista da empresa Meta

Basic algorithmic question. Improve time and space complexity.