566. 重塑矩阵

;(function () {
  /**
   * 566. 重塑矩阵
   * 在 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]]
   *
   * splice 会改变原数组， slice不会改变原数组
   */
  function matrixReshape(mat: number[][], r: number, c: number): number[][] {
    // 方法一： 暴力循环
    let rtnArr = []
    let flatArr = mat.flat()
    let len = flatArr.length
    // 给定的参数合理
    if (r * c === len) {
      while (flatArr.length > 0) {
        rtnArr.push(flatArr.splice(0, c))
      }
      return rtnArr
    } else {
      return mat
    }

    // 方法二：官方提供
    // const m = mat.length;
    // const n = mat[0].length;
    // if (m * n != r * c) {
    //     return mat;
    // }
    // const ans = new Array(r).fill(0).map(() => new Array(c).fill(0));
    // for (let x = 0; x < m * n; ++x) {
    //     //  ans[0][0] = mat[0][0]
    //     //  ans[0][1] = mat[0][1]
    //     //  ans[0][2] = mat[1][0]
    //     ans[Math.floor(x / c)][x % c] = mat[Math.floor(x / n)][x % n];
    // }
    // return ans;
  }

  console.log(
    matrixReshape(
      [
        [1, 2],
        [3, 4]
      ],
      1,
      4
    )
  )
  console.log(
    matrixReshape(
      [
        [1, 2],
        [3, 4]
      ],
      2,
      4
    )
  )
})()