Create convert-1d-array-into-2d-array.py

Author: kamyu104

Python/convert-1d-array-into-2d-array.py

@@ -0,0 +1,12 @@
+# Time:  O(m * n)
+# Space: O(1)
+
+class Solution(object):
+    def construct2DArray(self, original, m, n):
+        """
+        :type original: List[int]
+        :type m: int
+        :type n: int
+        :rtype: List[List[int]]
+        """
+        return [original[i:i+n] for i in xrange(0, len(original), n)] if len(original) == m*n else []