1170. 重塑矩阵

1170. 重塑矩阵

 

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

先给出一个由二维数组表示的矩阵，以及两个正整数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的矩阵。 所以输出原始矩阵。

注意事项

 

矩阵行和列的大小范围为[1, 100]。

给出的r和c都为正数。

 

 

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 here

         if (nums.length * nums[0].length < r * c) {

                return nums;

            }

            int[][] ret = new int[r][c];

            int i = 0, j = 0;

            for (int[] a : nums) {

                for (int b : a) {

                    ret[i][j] = b;

                    if (j < c-1) {

                        j++;

                    } else {

                        i++;

                        j = 0;

                    }

                }

            }

        return ret;

    }

}