LeetCode 566.重塑矩阵

题目：

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

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

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

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

示例 1：

输入：mat = [[1,2],[3,4]], r = 1, c = 4
输出：[[1,2,3,4]]

示例 2：

输入：mat = [[1,2],[3,4]], r = 2, c = 4
输出：[[1,2],[3,4]]

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/reshape-the-matrix

题解：

先理解一下题目中的参数是什么意思， matSize 表示数组行数， *matColSize 表示数组列数， *returnSize 表示返回数组的行数， **returnColumnSizes 表示返回数组的列数， *returnColumnSizes 表示返回数组每一行的元素个数（以数组形式表示）。
我们需要先将题目中给的二维数组映射成一个一维数组，再将这个一维数组映射成 r 行 c 列的一个二维数组。可以设出一个一维数组用来过渡，也可以直接按照下图进行转换：

/**
 * 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 m=matSize;
    int n=matColSize[0];
    if(m*n!=r*c)
    {
        *returnSize=matSize;
        *returnColumnSizes=matColSize;
        return mat;
    }
    *returnSize=r;
    *returnColumnSizes=malloc(sizeof(int*)*r);
    int **result=malloc(sizeof(int*)*r);
    for(int i=0;i<r;i++)
    {
        (*returnColumnSizes)[i]=c;
        result[i]=malloc(sizeof(int)*c);
    }
    for(int i=0;i<r*c;i++)
    {
        result[i/c][i%c]=mat[i/n][i%n];
    }
    return result;
}