leetcode 2022. 将一维数组转变成二维数组（java & python3）

祝大家元旦快乐！！！

java：

这道题和566.重塑矩阵相似

/  列数：可以取到每一行；%列数：可以取到每一列

class Solution {
    public int[][] construct2DArray(int[] original, int m, int n) {
        int l = original.length;
        
        if (l != m * n) {
            return new int[0][];
        }

        int [][] ans = new int[m][n];
        for(int i = 0; i < l; i++){
            ans[i / n][i % n] = original[i];
        }

        return ans;
    }
}

python3:

class Solution:
    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
        l = len(original)
        if l != m * n:
            return []
        ans = [[0 for _ in range(n)] for _ in range(m)]
        for i in range(l):
            ans[i // n][i % n] = original[i]
        return ans