566.重塑矩阵0ms

对于m*n大小的数组a，第i个元素的位置为a[i/n][i%n] (i/n为所在行，i%n为所在列)

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[][] result = new int[r][c];
        for(int i = 0; i < m*n; i++){
            result[i/c][i%c] = mat[i/n][i%n];
        }
        return result;
    }
}