注:本文的实现思路主要是基于JS(JavaScript),涉及到的一些函数都是JS中的自带函数
题目描述
在 MATLAB 中,有一个非常有用的函数 reshape ,它可以将一个 m x n 矩阵重塑为另一个大小不同(r x c)的新矩阵,但保留其原始数据。
给你一个由二维数组 mat 表示的 m x n 矩阵,以及两个正整数 r 和 c ,分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。
如果具有给定参数的 reshape 操作是可行且合理的,则输出新的重塑矩阵;否则,输出原始矩阵。
输入示例
输入:mat = [[1,2],[3,4]], r = 1, c = 4
输出:[[1,2,3,4]]
输入:mat = [[1,2],[3,4]], r = 2, c = 4
输出:[[1,2],[3,4]]
解题思路
这道题目使用最直接的数学逻辑进行求解即可。先把原始数组每一行一次拼接,形成一个一维数组,然后从这个一维数组中,依次截取c
个元素,放到一个数组中,然后追加到存放结果的 result
数组中即可。唯一需要注意的是,如果给定的r*c
比原数组的元素个数多,就不能实现,直接返回原数组即可。
var matrixReshape = function(mat, r, c) {
const row_num = mat.length
const col_num = mat[0].length
// 判断给定参数的 reshape 操作是可行且合理的,就是看变化前后元素个数是否一致
if (row_num * col_num !== r * c) {return mat}
// 创建一个临时数组,存在按行展开的一维数组
let temp = []
for (var i = 0; i < row_num; i++) {
// 把每一行的元素都追加到 temp 中
temp.push(...mat[i])
}
// 存放最后结果
let result = []
for (var i = 0; i < r; i++) {
// 每次根据指定的列数,对 temp 进行截取, 放到 result 中
result.push(temp.slice(i*c, (i+1)*c))
}
return result
};