数组-[5871] 将一维数组转变成二维数组

/**
5871. 将一维数组转变成二维数组
 */
class Solution {
    public int[][] construct2DArray(int[] original, int m, int n) {
        // 直接返回不符合的部分
        // original 数组元素数小于 m*n
        // original 数组元素数大于 m*n
        if(original.length < m * n || original.length > m *n)
            return new int[0][0];
        
        // 返回结果
        int[][] re = new int[m][n];
        
        for(int i = 0; i < m; i++){
            // 计算行数
            int row = n * i;
            for(int j = 0; j < n; j++){
                // 将对应位置放入re
                re[i][j] = original[row + j];
            }
        }

        return re;
    }
}