《LeetCode之每日一题》:277.重塑矩阵
作者:互联网
重塑矩阵
题目链接: 重塑矩阵
有关题目
在 MATLAB 中,有一个非常有用的函数 reshape ,
它可以将一个 m x n 矩阵重塑为另一个大小不同(r x c)的新矩阵,但保留其原始数据。
给你一个由二维数组 mat 表示的 m x n 矩阵,以及两个正整数 r 和 c ,
分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。
如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;
否则,输出原始矩阵。
提示:
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;
}
标签:mat,int,returnColumnSizes,矩阵,重塑,277,matColSize,LeetCode 来源: https://blog.csdn.net/MerryMaking7946/article/details/122675411