Amazon Interview Question

program k-nearest neighbour from draft