题目
在MATLAB中,有一个非常有用的函数 reshape,它可以将一个矩阵重塑为另一个大小不同的新矩阵,但保留其原始数据。
给出一个由二维数组表示的矩阵,以及两个正整数r和c,分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。
如果具有给定参数的reshape操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。
示例 1:
输入:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
输出:
[[1,2,3,4]]
解释:
行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。
示例 2:
输入:
nums =
[[1,2],
[3,4]]
r = 2, c = 4
输出:
[[1,2],
[3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。
注意:
给定矩阵的宽和高范围在 [1, 100]。
给定的 r 和 c 都是正数。
来源:力扣(LeetCode)
解题
模板
/**
* 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().
* 返回一个大小为*returnSize的数组。
*数组的大小以*returnColumnSizes数组的形式返回。
*注意:返回的数组和*columnSizes数组必须是错置的,假设调用方调用free()
*/
int** matrixReshape(int** nums, int numsSize, int* numsColSize, int r, int c, int* returnSize, int** returnColumnSizes){
}
分析题意
我们已知题目的要求就是对于我们输入的原二维数组进行一个变换,判断其是否能够变换成要求的r*c的二维数组,如果可以就输出转换后的二维数组,转换不成功就将原二维数组输出,所以我们需要对于原数组是否能够转换进行判断。
分步编码
初始化得到原数组的行数和列数
int m = numsSize;
int n = numsColSize[0];
得到后我们需要进行元素个数判断,来决定是否需要进行下一步的转换操作,如果不满足mn=rc,则直接返回原数组即可,同时对于returnSize和returnColumnSizes进行一个简单的和原数组的同步。
if (m * n != r * c) {
*returnSize = numsSize;
*returnColumnSizes = numsColSize;
return nums;
}
如果满足了元素个数合理就继续转换
首先对于新的r*c数组进行初始化
*returnSize = r;//r行
*returnColumnSizes = malloc(sizeof(int) * r);//初始化分配内存并返回指向可使用的内存地址大小为r个整型数据
int** ans = malloc(sizeof(int*) * r);//同上
for (int i = 0; i < r; i++) {
(*returnColumnSizes)[i] = c;//c列
ans[i] = malloc(sizeof(int) * c);//为每一行的初始指向一可用的内存地址大小为c个整型数据
}
初始化之后就可以对于两个数组之间进行转化;
转化方式很简单我们通过除法和模运算得到两者的对应位置
总元素个数是m*n个,行与列也都是从0开始的,所以x/n就是原二维数组中的第几行,x%n就是原二维数组中的第几列。同理,x/c就是新二维数组中的第几行,x%c就是新二维数组中的第几列
for (int x = 0; x < m * n; ++x) {
ans[x / c][x % c] = nums[x / n][x % n];
}
最后返回ans即可
完整代码
int** matrixReshape(int** nums, int numsSize, int* numsColSize, int r, int c, int* returnSize, int** returnColumnSizes) {
int m = numsSize;
int n = numsColSize[0];
if (m * n != r * c) {
*returnSize = numsSize;
*returnColumnSizes = numsColSize;
return nums;
}
*returnSize = r;
*returnColumnSizes = malloc(sizeof(int) * r);
int** ans = malloc(sizeof(int*) * r);
for (int i = 0; i < r; i++) {
(*returnColumnSizes)[i] = c;
ans[i] = malloc(sizeof(int) * c);
}
for (int x = 0; x < m * n; ++x) {
ans[x / c][x % c] = nums[x / n][x % n];
}
return ans;
}