leetcode.矩阵.566重塑矩阵-Java

1. 具体题目

给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示想要的重构的矩阵的行数和列数。重构后的矩阵需要将原始矩阵的所有元素以相同的行遍历顺序填充。如果具有给定参数的reshape操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

示例 1:　　输入: nums = [[1,2], [3,4]] r = 1, c = 4　　输出:  [[1,2,3,4]]　　解释: 行遍历nums的结果是 [1,2,3,4]。新的矩阵是 1 * 4 矩阵, 用之前的元素值一行一行填充新矩阵。

2. 思路分析

没有简便算法，只能将原矩阵中元素依次填入新矩阵中。对于新矩阵的行和列设置两层 for 循环，对于原矩阵，填入时通过索引判断是否需要换行。

3. 代码

 public int[][] matrixReshape(int[][] nums, int r, int c) {
         int n = nums.length;
         int m = nums[0].length;
         if(n * m != r * c) return nums;
         n = 0; m = 0;
         int[][] ans = new int[r][c];
        for(int i = 0; i < r; i++){
            for(int j = 0; j < c; j++){
                ans[i][j] = nums[n][m];
                m++;
                if(m == nums[0].length){
                     m = 0;
                     n++;
                }
            }
        }
        return ans;
     }