2021年2月17日 Leetcode每日一题：566. 重塑矩阵

重塑矩阵

1.题目描述

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

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

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

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

2.示例

示例 1:

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

示例 2:

输入: 
nums = 
[[1,2],
 [3,4]]
r = 2, c = 4
输出: 
[[1,2],
 [3,4]]
解释:
没有办法将 2 * 2 矩阵转化为 2 * 4 矩阵。 所以输出原矩阵。

3.读题

题目和示例描述的已经非常清楚了。这一题就是相当于，把一个矩阵按行接在一起形成一个数列，然后再用这个数列从头到尾按行去填充一个新的矩阵。
需要的注意的是，如果新矩阵不能被原来的矩阵的数字填满，那么就返回原矩阵。

4.思路

首先判断是否恰好能填充，如果不能则返回原矩阵。
如果能，则按行遍历原矩阵，取出每个数。同时，用两个标志位row和col分别代表新矩阵的行和列。每次填充一个值则新矩阵中的列坐标+1，当列坐标等于列长度时则换行，行坐标+1，列坐标归零。

5.代码

代码如下：

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        if(nums==null||nums[0]==null||r*c!=nums.length* nums[0].length) return nums;
        int[][] ans = new int[r][c];
        int row = 0;
        int col = 0;
        for(int i=0;i<nums.length;i++){
            for(int j=0;j<nums[0].length;j++){
                ans[row][col] = nums[i][j];
                row+=(col+1)/c;
                col = (col+1)%c;
            }
        }
        return ans;
    }
}