重塑矩阵

• 有关题目
• • 题解

题目链接： 重塑矩阵

有关题目

在 MATLAB 中，有一个非常有用的函数 reshape ，
它可以将一个 m x n 矩阵重塑为另一个大小不同（r x c）的新矩阵，但保留其原始数据。

给你一个由二维数组 mat 表示的 m x n 矩阵，以及两个正整数 r 和 c ，
分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。

如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；
否则，输出原始矩阵。

提示：

m == mat.length
n == mat[i].length
1 <= m, n <= 100
-1000 <= mat[i][j] <= 1000
1 <= r, c <= 300

题解

法一：二维数组的一维表示
参考官方题解

/**
 * Return an array of arrays of size *returnSize.
 * The sizes of the arrays are returned as *returnColumnSizes array.
 * Note: Both returned array and *columnSizes array must be malloced, assume caller calls free().
 */
int** matrixReshape(int** mat, int matSize, int* matColSize, int r, int c, int* returnSize, int** returnColumnSizes){
    int sz = matSize * matColSize[0];
    if (sz != r * c)
    {
        *returnSize = matSize;
        *returnColumnSizes = matColSize;
        return mat;
    }
    
    int i;
    *returnSize = r;
    *returnColumnSizes = (int*)malloc(sizeof(int) * r);
    int **ret = (int**)malloc(sizeof(int*) * r);
    
     for (i = 0; i < r; i++)
     {
         (*returnColumnSizes)[i] = c;
         ret[i] = (int*)malloc(sizeof(int) * c);
     }

     for (i = 0; i < sz; i++)//二维数组mat映射为一个一维数组，再将这个一维数组映射为[r, c]的矩阵
     {
         ret[i / c][i % c] = mat[i / matColSize[0]][i % matColSize[0]];
     }

     return ret;
}