566. 重塑矩阵(模拟)

 

 

　　最容易想到的就是创建一个新的一维数组,然后遍历一遍原数组nums,将其元素都按序填入新数组,最后在遍历一遍新创建的r*c的二维数组将一维数组的值按序填入.

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        if(nums.length==0||nums[0].length==0) return null;
        if(r*c!=nums.length*nums[0].length) return nums;
        int k=0,x=0; 6         int[][] res=new int[r][c];
        int[] temp=new int[nums.length*nums[0].length];
        for(int i=0;i<nums.length;i++){
            for(int j=0;j<nums[0].length;j++){
                temp[x++]=nums[i][j];
            }
        }
        for(int i=0;i<r;i++){
            for(int j=0;j<c;j++){
                res[i][j]=temp[k++];
            }
        }
        return res;
    }
}

 　　稍微比前面那种暴力快一点的方法

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        if(nums.length==0||nums[0].length==0) return null;
        if(r*c!=nums.length*nums[0].length) return nums;
        int k1=0,k2=0;
        int[][] res=new int[r][c];
        for(int[] ref:nums){
            for(int val:ref){
                res[k1][k2++]=val;
                if(k2==c){ //一行填满,转换到下一行
                    k1++;
                    k2=0;
                }
            }
        }
        return res;
    }
}