【leetcode】二维网格迁移

int** shiftGrid(int** grid, int gridSize, int* gridColSize, int k, int* returnSize, int** returnColumnSizes){
    int** arr = (int**)calloc(gridSize,sizeof(int*));
    int i,j,row,col;
    for (i=0; i<gridSize; i++)
    {
        int* rowAddr = (int*)calloc(gridColSize[i],sizeof(int));
        arr[i] = rowAddr;
    }
    for (i=0; i<gridSize; i++)
    {    
        for (j=0; j<gridColSize[i]; j++)
        {
            row = (gridColSize[i]*i+j+k)/gridColSize[i]%gridSize;
            col = (gridColSize[i]*i+j+k)%gridColSize[i];
            arr[row][col] = grid[i][j];
        }
        (*returnColumnSizes)[i] = gridColSize[i];
    }
    *returnSize = gridSize;
    return arr;
}