566_javascript.js

// 在MATLAB中，有一个非常有用的函数 reshape，它可以将一个矩阵重塑为另一个大小不同的新矩阵，但保留其原始数据。

// 给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。

// 重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。

// 如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

// 示例 1:

// 输入:
// nums =
// [[1,2],
//  [3,4]]
// r = 1, c = 4
// 输出:
// [[1,2,3,4]]
// 解释:
// 行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。
// 示例 2:

// 输入:
// nums =
// [[1,2],
//  [3,4]]
// r = 2, c = 4
// 输出:
// [[1,2],
//  [3,4]]
// 解释:
// 没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。
// 注意：

// 给定矩阵的宽和高范围在 [1, 100]。
// 给定的 r 和 c 都是正数。

/**
 * @param {number[][]} nums
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */

// 对于数组m行n列数组 元素[i,j] 可以衍变为 i*n + j
// 所以 得出等式
//
//  i = x / n    x为二维数组转变为一维数组的第x个数
//  j = x % n

var matrixReshape = function (nums, r, c) {
  let m = nums.length;
  let n = nums[0].length;
  if (n * m !== r * c) return nums;
  let res = Array.from(new Array(r), arr => new Array(c));

  for (let i = 0; i < m * n; i++) {
    // 带入以上等式
    res[Math.floor(i / c)][i % c] = nums[Math.floor(i / n)][i % n];
  }
  return res;
};