Pergunta de entrevista da empresa Uber

Solve this problem given an input and output.