Citadel Interview Question

Implement K nearest neighbour