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;

     }

巴特西

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

最新文章

热门文章