Walmart interview question

Code K means from scratch