leetcode刷题记录-566.重塑矩阵[java,数组]

题目

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

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

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

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/reshape-the-matrix
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

我的代码

class Solution {
    public int[][] matrixReshape(int[][] nums, int r, int c) {
        if (nums.length*nums[0].length < r*c) {return nums;}
        int[] allin = new int[nums.length*nums[0].length];
        int[][] ans = new int[r][c];
        int count = 0;
        for (int i=0; i<nums.length; i++) {
            for (int j=0; j<nums[0].length; j++) {
                allin[count++] = nums[i][j];
            }
        }
        count = 0;
        for (int i=0; i<r; i++){
            for (int j=0; j<c; j++) {
                ans[i][j] = allin[count++];
            }
        }
        return ans;
    }
}

小结

本来我想用Arrays.copyOfRange实现allin的拆分，但是总是提示数组越界，看来看去也没看出哪里越界，没办法还是老实人双层for循环进行ans的填装。
希望在大佬们的答案里找一个Arrays.copyOfRange参考学习一下，但也未能如愿。