算法-重塑矩阵

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

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

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

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

来源：力扣（LeetCode）
 

一、老老实实一个一个双循环读取存入

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
         int x = mat.length;
        int y = mat[0].length;
        int[][] arr = new int[r][c];
        int p = 0;
        int q = 0;
        if (x * y != r * c) return mat;

        for (int i = 0; i < x; i++) {
            for (int j = 0; j < y; j++) {
                arr[p][q] =  mat[i][j];
                if(q == c-1){
                    p++;
                    q = 0;
                }else {
                    q++;
                }
            }
        }

        return arr;
    }
}

二、单循环

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
         int x = mat.length;
        int y = mat[0].length;
        int[][] arr = new int[r][c];
        int p = 0;
        int q = 0;
        int z = -1;
        if (x * y != r * c) return mat;

          for (int i = 0; i < x * y; i++) {
            if (i % y == 0) {
                z++;
            }
            arr[p][q] = mat[z][i % y];

            if (q == c - 1) {
                p++;
                q = 0;
            } else {
                q++;
            }
        }


        return arr;
    }
}

是件复杂度其实没有变化，但是发现内存消耗反而上升了

三、再精简一点，本质是一样的

class Solution {
    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;
    }
}