开启掘金成长之旅!这是我参与「掘金日新计划 · 12 月更文挑战」的第6天,点击查看活动详情
描述
在 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 <= mati <= 1000
- 1 <= r, c <= 300
解析
根据题意,看了官方题解和一些大神的思路,谈谈对本题的理解。
-
首先计算原二维数组的长度,和每一位元素长度,他们的乘积决定了自定义的两个变量是否可以实现要求
-
根据输入的值乘积是否等于原本的行列乘积,不符合要求原样返回
-
如果符合要求,先根据自定义两个变量,根据入参来声明二维数组。
-
通过for循环,n2控制mat的取值范围,声明初始值为0的两个指针,用n2约束二维数组中一位数组的长度。
Java解题过程如下:
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[][] d = new int[r][c];
int x1 = 0;
int x2 = 0;
for (int i = 0; i < r; i++) {
for (int j = 0; j < c; j++) {
if (x2 == n2) {
x1 += 1;
x2 = 0;
}
int index1 = x1;
int index2 = x2;
d[i][j] = mat[index1][index2];
x2++;
}
}
return d;
}
}
运行结果:
执行结果:通过
执行用时:3 ms, 在所有 Java 提交中击败了30.35%的用户
内存消耗:52.4 MB, 在所有 Java 提交中击败了94.78%的用户