1. 程式人生 > >[LeetCode javaScript] 566. 重塑矩陣

[LeetCode javaScript] 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 都是正數。

/**
 * @param {number[][]} nums
 * @param {number} r
 * @param {number} c
 * @return {number[][]}
 */
var matrixReshape = function(nums, r, c) {
    var dd=[];
    for(var i =0;i<r;i++){
        dd[i]=[];
    }
    var x=0;
    var y=0;
    for (var i=0;i<nums.length;i++){
        for (var j=0;j<nums[i].length;j++){
                if(y<c){
                    dd[x][y]=nums[i][j];
                    y++;
                }else{
                    y=0;
                    x++;
                    dd[x][y]=nums[i][j];
                    y++;
                }
        }
    }
    //無法組成目標陣列,返回原陣列
    if(dd[dd.length-1][0]==null){
        return nums;
    }
    return dd;
};