566. Reshape the Matrix重塑矩阵Java

在 MATLAB 中，有一个方便的函数调用reshape，它可以将m x n矩阵重塑为具有不同大小的新矩阵，并r x c保留其原始数据。
给你一个m x n矩阵mat和两个整数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]]

约束：
m == mat.length
n == mat[i].length
1 <= m, n <= 100
-1000 <= mat[i][j] <= 1000
1 <= r, c <= 300

思路

1. 题目中说如果给的参数可运行就输出新的矩阵, 否则就就输出原矩阵即只有m×n=r×c时才可以运行
   if (r * c != mat.length * mat[0].length) { return mat; }
2. 如果程序可运行则初始化一个r行c列的数组, 用于存放数并输出
3. 初始化i=0,j=0为了遍历给定数组mat中的数
4. 遍历这个新的数组的行和列(两个for循环)
5. 当j=n时证明mat数组当前行的最后一个数已经遍历完, 将i++,j=0
6. 将mat中的数存到输出数组中

class Solution {
    public int[][] matrixReshape(int[][] mat, int r, int c) {
        if (r * c != mat.length * mat[0].length) {
            return mat;
        }
        int[][] res = new int[r][c];
        int i = 0, j = 0;
        for (int row = 0; row < r; row++) {
            for (int column = 0; column < c; column++) {
                if (j == mat[0].length) {
                    i++;
                    j = 0;
                }
                res[row][column] = mat[i][j++];
            }
        }
        return res;
    }
}

时间复杂度O(r×c)