566 重塑矩阵

在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 都是正数。

题解

1. 思路是先把二维数组先转为一维数组。然后遍历行数的值，每次根据列数的值去截取数据，如果在截取的过程中，获取不到数据，那么说明失败，则返回原数组。

TIP

执行用时：108 ms, 在所有 JavaScript 提交中击败了 71.2% 的用户
内存消耗：42.9 MB, 在所有 JavaScript 提交中击败了 61.9% 的用户

/**
 * @param {number[][]} nums
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */
var matrixReshape = function(nums, r, c) {
    let copyNums = [].concat(...nums) // 把数组转为一维数组
    let result = []
    
    // r 的值决定了需要插入几个第二维数组
    while (r--) {
        let item = copyNums.splice(0, c) // 每次截取固定的数据
        if(item.length) {
            // 如果截取到了数据，就存入
            result.push(item) 
        } else {
            // 截取不到说明失败，则返回原数组
            return nums
        } 
    }

    return result
};

• 时间复杂度： `O(n)`， 一个 while循环，r 控制遍历次数
• 空间复杂度： `O(n)`，使用了一个额外的数组
• 最好情况： r的值非常小，遍历的次数就越小
• 最坏情况： r的值非常大，那么需要遍历多次