Pregunta de entrevista de TuSimple

algorithm related question, in python