重塑矩阵（matlab reshape函数原理）

Posted 2021-09-06 秦枫-_-

简单来说就是同样容量的情况下才能转换，否则就输出原矩阵
(i,j)对应了矩阵中第i*n+j个数，对应ans中idx/c，idx%c

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
    int m=mat.length,n=mat[0].length;
    if(m*n!=r*c)return mat;
      int [][]ans=new int[r][c];
   for(int i=0;i<m;i++){//(i,j)对应了矩阵中第i*n+j个数，对应ans中idx/c，idx%c
    for(int j=0;j<n;j++){
        ans[(i*n+j)/c][(i*n+j)%c]=mat[i][j];
    }
   }
  
    return ans;
    }
}