【力扣-数据结构】【第 4 天】566. 重塑矩阵

标题	566. 重塑矩阵
难度	简单
天数	第4天,第1/2题
数据结构	数组

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

给你一个由二维数组 mat 表示的 m x n 矩阵，以及两个正整数 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

来源：力扣（LeetCode）
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

1. 本题一个必要条件就是m * n = r * c 不满足直接返回原始矩阵

	int m = mat.length;
    int n = mat[0].length;
    //不能转换返回原始矩阵
    if(m * n != r * c){
        return mat;
    }

2. 创建一个r*c的二维数组

	//创建需要转换的矩阵 r * c
	int[][] res = new int[r][c];

3. 记录一维矩阵赋值位置 ，记录二维矩阵赋值位置,初始都是0

 	//记录一维矩阵赋值位置
	int rIndex = 0; 
	//记录二维矩阵赋值位置
	int cIndex = 0;

4. 双层for循环遍历m*n 的二维数组mat取出元素

	for(int i = 0; i < m;i++){
		for(int j = 0 ;j < n;j++){
			? = mat[i][j];
		}
	}

5. res 装满一行就换到下一行，并且从0开始赋值

	if(cIndex >= c){
		rIndex++;
		cIndex = 0;
	}

6. 开始赋值，然后将位置切换到下一个位置

	//赋值
	res[rIndex][cIndex] = mat[i][j];
	//赋值后，将位置移动到下一个位置
	cIndex++;

最后完整代码：

class Solution {
    //数据结构   第 4 天 1/2 数组
    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;
        }
        //创建需要转换的矩阵 r * c
        int[][] res = new int[r][c];
        //记录一维矩阵赋值位置
        int rIndex = 0; 
        //记录二维矩阵赋值位置
        int cIndex = 0;
        for(int i = 0; i < m;i++){
            for(int j = 0 ;j < n;j++){
                //如果一横排已满，切换到下一个横排，并从0开始赋值
                if(cIndex >= c){
                    rIndex++;
                    cIndex = 0;
                }
                //赋值
                res[rIndex][cIndex] = mat[i][j];
                //赋值后，将位置移动到下一个位置
                cIndex++;
            }
        }
        return res;
    }
}