Domanda di colloquio di Amazon

Leetcode question with two parts