《LeetCode之每日一题》:277.重塑矩阵

重塑矩阵


题目链接: 重塑矩阵

有关题目

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

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

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

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

《LeetCode之每日一题》:277.重塑矩阵
《LeetCode之每日一题》:277.重塑矩阵

提示:

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;
}

《LeetCode之每日一题》:277.重塑矩阵

上一篇:面向对象编程C++(基本知识点)


下一篇:LeetCode 107. 二叉树的层序遍历 II