【Leetcode】2022. Convert 1D Array Into 2D Array

题目地址：

https://leetcode.com/problems/convert-1d-array-into-2d-array/

给定一个一维数组$A$，将其转为$m\times n$的二维数组。

代码如下：

public class Solution {
    
    
    public int[][] construct2DArray(int[] original, int m, int n) {
    
    
        if (m * n != original.length) {
    
    
            return new int[0][0];
        }
        
        int[][] res = new int[m][n];
        for (int i = 0; i < original.length; i++) {
    
    
            res[i / n][i % n] = original[i];
        }
        
        return res;
    }
}

时间复杂度$O(mn)$，空间$O(1)$。