Walmart Interview Question

Code K means from scratch