Domanda di colloquio di Uber

Solve this problem given an input and output.