在 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]]
思路:那么题目需要我们做的事情相当于:将二维数组nums 映射成一个一维数组;将这个一维数组映射回 r 行 cc列的二维数组。我们当然可以直接使用一个一维数组进行过渡,但我们也可以直接从二维数组nums 得到 r 行 cc列的重塑矩阵:
设 nums 本身为 m 行 n 列,如果 mn = rc那么二者包含的元素个数不相同,因此无法进行重塑;否则,对于 x∈[0,mn),第 x 个元素在nums 中对应的下标为 (x / n,x % n),而在新的重塑矩阵中对应的下标为(x / c,x % c)。我们直接进行赋值即可。
int R=mat.length;
int C = mat[0].length;
if(R*C!=r*c){
return mat;
}
int[][] mat2 = new int[r][c];
for (int i = 0; i < R * C; i++) {
mat2[i / c][i % c] = mat[i / C][i % C];
}
System.out.println(mat2.toString());
return mat2;