In MATLAB, there is a very useful function called ‘reshape’, which can reshape a matrix into a new one with different size but keep its original data.
You’re given a matrix represented by a two-dimensional array, and two positive integers r and c representing the row number and column number of the wanted reshaped matrix, respectively.
The reshaped matrix need to be filled with all the elements of the original matrix in the same row-traversing order as they were.
If the ‘reshape’ operation with given parameters is possible and legal, output the new reshaped matrix; Otherwise, output the original matrix.
在MATLAB中,有一个非常有用的函数叫做“reshape”,它可以将矩阵重新形成一个不同大小但保持其原始数据的新矩阵。
给出一个由二维数组表示的矩阵,两个正整数r和c分别表示所需重组矩阵的行号和列号。
重新组合的矩阵需要以与它们相同的行遍历顺序填充原始矩阵的所有元素。
如果具有给定参数的“重塑”操作是可行且合法的,则输出新的重构矩阵; 否则,输出原始矩阵。
Example 1:
Input:
nums =
[[1,2],
[3,4]]
r = 1, c = 4
Output:
[[1,2,3,4]]
Explanation:
The row-traversing of nums is [1,2,3,4]. The new reshaped matrix is a 1 * 4 matrix, fill it row by row by using the previous list.
Example 2:
Input:
nums =
[[1,2],
[3,4]]
r = 2, c = 4
Output:
[[1,2],
[3,4]]
Explanation:
There is no way to reshape a 2 * 2 matrix to a 2 * 4 matrix. So output the original matrix.
Note:
The height and width of the given matrix is in range [1, 100].
The given r and c are all positive.
思路:
根据给出的 nums 中的元素总数,判断给出的 行数和列数的乘积是否合法,
- 不合法返回原矩阵
- 若合法,则将nums 中的所有元素都放到一个新的数组中,再重新放到另一个新数组中。。。