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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

• 这就是个模拟类型的题，我以为它会很简单。
• 但是做着做着，我就发现returnColumnSizes这个双重指针该怎么初始化，还有二维数组该如何初始化。
• 这两个问题把我卡了很久，最不得不自己去百度了一下。

代码

二维数组的初始化

int** num=(int**)malloc(r*sizeof(int*));
    for(int i=0;i<r;i++)
    {
        num[i]=(int*)malloc(c*sizeof(int));
    }

returnColumnSizes的初始化

*returnColumnSizes=(int*)malloc(r*sizeof(int));
    for(int i=0;i<r;i++)
    {
        returnColumnSizes[0][i]=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){
    if(matSize*(*matColSize)!=r*c||matSize==r&&*matColSize==c)
    {
        *returnSize=matSize;
        *returnColumnSizes=(int*)malloc(matSize*sizeof(int));
        for(int i=0;i<matSize;i++)
        {
            returnColumnSizes[0][i]=*matColSize;
        }
        return mat;
    }
    int** num=(int**)malloc(r*sizeof(int*));
    for(int i=0;i<r;i++)
    {
        num[i]=(int*)malloc(c*sizeof(int));
    }
    int numr=0,numc=0;//记录新矩阵的行和列；
    for(int i=0;i<matSize;i++)
    {
        for(int j=0;j<*matColSize;j++)
        {
            num[numr][numc++]=mat[i][j];
            if(numc==c)
            {
                numc=0;
                numr++;
            }
        }
    }
    *returnSize=r;
    *returnColumnSizes=(int*)malloc(r*sizeof(int));
    for(int i=0;i<r;i++)
    {
        returnColumnSizes[0][i]=c;
    }
    return num;
}

注释

• returnColumnSizes的作用其实是用来记录新数组每一行的列数的。所以它和新数组num的初始化会有写不一样。