题目:
在 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;
}