20220101|力扣|每日一题|2022

[难度：简单] 点我做题去

    public int[][] construct2DArray(int[] original, int m, int n) {
        int[][] res = new int[m][n];
        // 判断是否符合天剑，不符合直接返回空[]
        if (original.length != (m * n)) {
            return new int[0][];
        }
        // 记录在数组 original 中的位置
        int oriIndex = 0;
        // 记录写入列的位置
        int mIndex = 0;
        // 记录写入行的位置
        int nIndex = 0;
        // while写入
        while (mIndex < m) {
            while (nIndex < n) {
                res[mIndex][nIndex++] = original[oriIndex++];
            }
            mIndex++;
            nIndex = 0;
        }
        return res;
    }

    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
        res = []
        length = len(original)
        if m * n != length:
            return []
        tmp = []
        for i in range(length):
            tmp.append(original[i])
            if len(tmp) == n:
                res.append(tmp)
                tmp = []
        return res