力扣题目——566. 重塑矩阵

注：本文的实现思路主要是基于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
};