2022. Convert 1D Array Into 2D Array

class Solution:
    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
        idealSize = len(original)

        if m * n != idealSize:
            return [] 

        arr = [[0 for _ in range(n)] for _ in range(m)]

        index = 0
        for i in range(m):
            for j in range(n):
                if index < idealSize:
                    arr[i][j] = original[index]
                    index += 1

        return arr