Anonymous interview question

K closest elements - medium question of leetcode