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

kamyu104 · Oct 3, 2021 · 5cd6311 · 5cd6311
1 parent 83872b0
commit 5cd6311
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/Python/convert-1d-array-into-2d-array.py b/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 []