Pregunta de entrevista de Uber

Easy coding question. Hard case study