def col_indices(n, square_length):
    """ Returns the one-dimensional indices of all the cells belonging to the
		same column as index `n`.
	"""
    j = n % square_length
    return range(j, square_length * (square_length - 1) + j + 1, square_length)
