领扣LintCode算法问题答案-1170. 重塑矩阵

领扣LintCode算法问题答案-1170. 重塑矩阵

1170. 重塑矩阵

描述

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

先给出一个由二维数组表示的矩阵，以及两个正整数r和c，分别表示所需重新整形矩阵的行数和列数。

重新形成的矩阵需要用原始矩阵的所有元素以相同的行遍历顺序填充。

如果使用给定参数的“重塑”操作是可能且合法的，则输出新的重新整形矩阵; 否则，输出原始矩阵。

• 矩阵行和列的大小范围为[1, 100]。
• 给出的r和c都为正数。

样例 1：

输入: 
nums = 
[[1,2],[3,4]]
r = 1, c = 4
输出: 
[[1,2,3,4]]
解释:
行遍历的顺序为 [1,2,3,4]. 新给出的矩阵大小为1 * 4, 根据前面给出的列表按行遍历即可。

样例 2：

输入: 
nums = 
[[1,2],[3,4]]
r = 2, c = 4
输出: 
[[1,2],[3,4]]
解释:
无法将一个2 * 2的矩阵整形称为一个2 * 4的矩阵。 所以输出原始矩阵。

题解

public class Solution {
    /*** @param nums: List[List[int]]* @param r: an integer* @param c: an integer* @return: return List[List[int]]*/public int[][] matrixReshape(int[][] nums, int r, int c) {
    // write your code hereif (nums == null) {
    return null;}int maxR = nums.length;if (maxR == 0) {
    return nums;}int maxC = nums[0].length;if (maxC == 0) {
    return nums;}if (maxR * maxC != r * c) {
    return nums;}int[][] ret = new int[r][c];int ri = 0;int ci = 0;for (int i = 0; i < maxR; i++) {
    for (int j = 0; j < maxC; j++) {
    ret[ri][ci] = nums[i][j];if (ci + 1 < c) {
    ci++;} else {
    ri++;ci = 0;}}}return ret;}
}

原题链接点这里

鸣谢

非常感谢你愿意花时间阅读本文章，本人水平有限，如果有什么说的不对的地方，请指正。
欢迎各位留言讨论，希望小伙伴们都能每天进步一点点。