Leetcode 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 矩陣。 所以輸出原矩陣。

來源：力扣（LeetCode）
鏈接：https://leetcode-cn.com/problems/reshape-the-matrix
著作權歸領釦網絡所有。商業轉載請聯繫官方授權，非商業轉載請註明出處。

 

/**
 * @param {number[][]} nums
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */
var matrixReshape = function(nums, r, c) {
    let res = [];
    let len1 = nums.length, len2 = nums[0].length;
    if(r * c !== len1 * len2) {
        return nums;
    } else {
        let arr = [].concat.apply([], nums);
        for(let i = 0; i < r; i++) {
            res[i] = [];
            for(let j = 0; j < c; j++) {
                res[i][j] = arr.shift();
            }
        }
    }
    return res;
};