提示:文章写完后,目录可以自动生成,如何生成可参考右边的帮助文档
问题描述
在MATLAB中,有一个非常有用的函数 reshape ,它可以将一个(m * n)的矩阵重塑为另一个大小不同的新矩阵(r * c),但保留其原始数据。
给你一个由二维数组 mat 表示的(m * n)矩阵,以及两个正整数 r 和 c ,分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。
如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。
来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/reshape-the-matrix
一、测试用例
示例一:
输入:mat = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]
示例二:
输入:mat = [[1,2],[3,4]], r = 2, c = 4
输出:[[1,2],[3,4]]
二、解法思路
见到这一题的时候首先想到了将二维数组转化为一维数组,再根据题目要求的行列数转化为新的二维数组。由于题目中提到可能会出现新矩阵与原矩阵元素数不相等的情况,因此添加元素数量的判断,如果相等则进行矩阵重塑,不相等则返回原来的矩阵。
个人认为这道题的一个问题就是一见到题目所给函数中的几个参数会使刚开始接触算法的新人难以理解,我就根据我自己的理解来说一下这几个参数的含义。
题目所给的函数定义如下:
/**
* 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){}
mat即为使用二级指针所定义的二维数组用于表示原矩阵,matSize为记录mat的长度,而数组matColSize为二级指针mat中每一个一级指针所指向的数组的长度,即矩阵每一行有多少列。整个二维数组由mat所指向的matSize个一维数组所组成。图示如下:
参数中的r为新矩阵的行数,c为新矩阵的列数,returnSize为所返回的二级指针的长度,相当于matSize,returnColumnSizes则相当于matColSize。
三、代码
代码如下:
int** matrixReshape(int** mat, int matSize, int* matColSize, int r, int c, int* returnSize, int** returnColumnSizes){
int m = matSize; //matSize为二级指针mat中的空间个数,即为二维数组mat的行数
int n = matColSize[0]; //matColSize为存储mat的一级指针指向的数组的长度,即为mat的每一行有多少列
if(r*c != m*n){ //如果矩阵中的元素数量不相等则无法重塑矩阵
*returnSize = matSize;
*returnColumnSizes = matColSize;
return mat;
}
int *d = (int *)malloc(sizeof(int)*(m*n)); //定义一维数组将mat一维化
*returnSize = r; //returnSize记录新的二级指针的空间个数,即新矩阵的行数
*returnColumnSizes = malloc(sizeof(int)*r); //returnColumnSizes存储每一行的列数,一共有r行,故有r个空间
int flag=0;
for(int i=0;i<m;i++){ //二维数组一维化
for(int j=0;j<n;j++){
d[flag] = mat[i][j];
flag++;
}
}
int **l = malloc(sizeof(int *)*r); //定义新的二级指针当作二维数组
flag = 0;
for(int i=0;i<r;i++){
l[i] = malloc(sizeof(int)*c); //l为二级指针所指的一维数组,保存新矩阵的每一行的元素,每一行有c列
(*returnColumnSizes)[i]=c; //returnColumnSizes存储第i行的列数
for(int j=0;j<c;j++){
l[i][j] = d[flag]; //一维数组二维化
flag++;
}
}
return l;
}
总结
我这样的解法并不是最优解,也与官方给的解法不同,但是个人认为相较于其他解法可能搞好理解吧(大概)。如果内容有问题请各位大佬指正,也欢迎大家在评论区讨论。最后给出我这种解法在力扣上跑的结果截图供大家参考。