Pergunta de entrevista da empresa NVIDIA

Coding question on Hacker Rank