题意
在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 都是正数。
解题思路
此题简单题,一些思路放入代码中,代码中含文字说明
代码演示
class Solution {
public int[][] matrixReshape(int[][] nums, int r, int c) {
int han=nums.length;
int lie=nums[0].length;
//不满足条件,不可以reshape的情况
if(han*lie!=r*c)
return nums;
//重新定义一个二维数组,用于存放reshape的值
int[][] reshape=new int[r][c];
//定义两个变量,rr代表reshape中的行,rc代表reshape中的列,初始值都是0
int rr=0,rc=0;
//双重循环,从nums中取数
for (int i=0;i<han;i++)
{
for (int j=0;j<lie;j++)
{
//将nums中的数取出来,按照要求放入到reshape中
reshape[rr][rc]=nums[i][j];
//以行的方式存储,reshape当前行右移
rc++;
//当一行满的时候,列向下移,当前行的指针重新归0
if(rc==c)
{
rr++;
rc=0;
}
}
}
return reshape;
}
}
效果
info
解答成功:
执行耗时:1 ms,击败了100.00% 的Java用户
内存消耗:39.7 MB,击败了25.85% 的Java用户