Pergunta de entrevista da empresa TUQ

Solve this LeetCode Easy, then make it arbitrary.