TuSimple interview question

implement K means algorithm