Robinhood Software Engineer Entry Level interview questions

One of the question from assessment: Given a grid of nxm with black and white squares (like chess board). Each position of the grid has a number. For every KxK matrix sort the numbers on the black squares and white squares separately and reassign the number on the squares.