力扣小白刷题之566题重塑矩阵

题目描述

在MATLAB中，有一个非常有用的函数 reshape，它可以将一个矩阵重塑为另一个大小不同的新矩阵，但保留其原始数据。
给出一个由二维数组表示的矩阵，以及两个正整数 r 和 c，分别表示想要的重构的矩阵的行数和列数。
重构后的矩阵需要将原始矩阵的所有元素以相同的 行遍历顺序 填充。
如果具有给定参数的 reshape 操作是可行且合理的，则输出新的重塑矩阵；否则，输出原始矩阵。

思路

• 如果原始矩阵中的元素数量不等于所得矩阵中的元素数量，则不可能形成所得矩阵。

参考自：https://leetcode-cn.com/problems/reshape-the-matrix/solution/566java-zhao-gui-lu-jiu-hao-by-ustcyyw/

• 无法完成 reshape 的意思就是，原矩阵元素无法恰好填充指定行数列数的新矩阵。所以当 r * c != nums.length * nums[0].length 时，返回原矩阵。
• 遍历原矩阵，第 0，1，2…，c - 1 个元素放入第一行(索引为 0)，其中第 0 个元素放入第一列，…，第 c - 1 个元素放入第 c - 1列。第 c，c + 1，…，2c - 1个元素放入第二行(索引为 1)…
• 以此类推，所以需要一个变量 count 来表示当前是第几个元素，因为索引是从 0 开始的，所以 count = 0 表示第一个元素。第 count 个元素应该放入 count / c 行，count % c 列。
• 遍历使用嵌套的 for-each 语句即可，每添加一个元素进入新矩阵中，使得计算 count++。

代码