LeetCode 数据结构—重塑矩阵

 

对于一个二维数组映射成另一个二维数组，我们可以将二维数组先映射成一个一维数组，然后再将一维数组逐个转为二维数组。

    public int[][] matrixReshape(int[][] mat, int r, int c) {
        int m = mat.length;
        int n = mat[0].length;
        if (m * n != r * c) {
            return mat;
        }

        int[][] ans = new int[r][c];
        for (int x = 0; x < m * n; ++x) {
            ans[x / c][x % c] = mat[x / n][x % n];
        }
        return ans;

    }