持续创作，加速成长！这是我参与「掘金日新计划 · 10 月更文挑战」的第16天，点击查看活动详情
题目描述
原题链接 :
在 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
思路分析
- 先算出原二维数组的长度，和每一位元素中长度，他们的乘积决定了自定义的两个变量是否可以实现要求
- 不符合要求原样返回
- 符合要求，先根据自定义两个变量，声明一下结果集，走到这一步，别人给几行几列，我们就声明几行几列的二维数组，不需要顾忌。
- 自定义的两个变量是两次for循环的次数，也是结果集对接的位置，如果采用for-for，难在取元素组mat的值，目前声明的变量中，n2明显可以控制mat的取值范围，声明初始值为0的两个指针，用n2约束二维数组中一位数组的长度，避免取值越界即可。
AC 代码
class Solution {
public int[][] matrixReshape(int[][] mat, int r, int c) {
// 可以理解成 行 和 列的值
int n1 = mat.length;
int n2 = mat[0].length;
// 输入的值乘积不等于原本的行列乘积，则说明无效（做不到），返回原样即可
if (n1 * n2 != r * c) {
return mat;
}
// 如果可以实现，按照输入的两个自定义变量，声明二维数组(结果集)，比如int[2][2] -> [[x,x],[x,x]]; int[1][3] -> [[x,x,x]]
int[][] d = new int[r][c];
// 循环从下标0开始，因为乘积中行列都是1开始的，这个乘积就像length，循环的时候，我们依旧按照length结束的写法
// for (int i = 0; i < n1 * n2; i++) {
// d[i / c][i % c] = mat[i / n2][i % n2];
// }
int x1 = 0;
int x2 = 0;
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
// 只有n2能控制mat不越界....
if (x2 == n2) {
x1 += 1;
x2 = 0;
}
int index1 = x1;
int index2 = x2;
d[i][j] = mat[index1][index2];
x2++;
}
}
return d;
}
}
