566. 重塑矩阵

题目

在 MATLAB 中，有一个非常有用的函数 reshape ，它可以将一个 m x n 矩阵重塑为另一个大小不同（r x c）的新矩阵，但保留其原始数据。

给你一个由二维数组 mat 表示的 m x n 矩阵，以及两个正整数 r 和 c ，分别表示想要的重构的矩阵的行数和列数。

重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。

如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例 1：

输入：mat = [[1,2],[3,4]], r = 1, c = 4
输出：[[1,2,3,4]]
示例 2：

输入：mat = [[1,2],[3,4]], r = 2, c = 4
输出：[[1,2],[3,4]]

代码

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        if(mat==null)
        return null;
        int m=mat.length;
        int n=mat[0].length;
        if(m*n!=r*c)
        return mat;
        int num=0;
        int[][] array=new int[r][c];
        int rm=0;int cm=0;
        for(int i=0;i<m;i++)
        for(int j=0;j<n;j++)
        {
            if(num>=c)
            {   num=0;
                rm++;
                array[rm][num++] =mat[i][j];
                System.out.printf("rm:%d  cm:%d \n",rm,cm)   ; 
            }else
            {
                array[rm][num++]=mat[i][j];
                System.out.printf("rm:%d  cm:%d \n",rm,cm)   ; 
            }
        }
        /*
        for(int i=0;i<r;i++)
        for(int j=0;j<c;j++)
        System.out.println(" "+String.valueOf(array[i][j])+" ");*/
        return array;
    }
}

反思

这个也不错，没有占用多余的空间。

还有一个好的思路：

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        if(mat==null)
        return null;
        int m=mat.length;
        int n=mat[0].length;
        if(m*n!=r*c)
        return mat;
        int[][] array=new int[r][c];
        int index=0;
        for(int i=0;i<r;i++)
        for(int j=0;j<c;j++)
        {
            array[i][j]=mat[index/n][index%n];
            index++;
        }
        /*
        for(int i=0;i<r;i++)
        for(int j=0;j<c;j++)
        System.out.println(" "+String.valueOf(array[i][j])+" ");*/
        return array;
    }
}

数与下标的对应关系。