LC 2022 [E] Convert 1D Array Into 2D Array - ALawliet/algorithms GitHub Wiki

class Solution:
    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
        if m * n != len(original):
            return []
        
        ROWS = m
        COLS = n
        res = deque()
        
        for r in range(ROWS):
            row = deque()
            for c in range(COLS):
                row.appendleft(original.pop())
            res.appendleft(row)
        
        return res

class Solution:
    def construct2DArray(self, original: List[int], m: int, n: int) -> List[List[int]]:
        ans = []
        if len(original) == m*n: 
            for i in range(0, len(original), n): 
                ans.append(original[i:i+n])
        return ans