Citadel interview question

Implement K nearest neighbour