Pergunta de entrevista da empresa Meta

Leetcode style question, nothing tricky