Airbnb interview question

write code to execute KNN