Time: 56 ms (71.26%), Space: 17.8 MB (98.22%) - LeetHub

Author: hogan-tech

1580-shuffle-the-array/1580-shuffle-the-array.py

@@ -0,0 +1,23 @@
+# time complexity: O(n)
+# space complexity: O(1)
+from typing import List
+
+
+class Solution:
+    def shuffle(self, nums: List[int], n: int) -> List[int]:
+        result = []
+        for i in range(n):
+            result.append(nums[i])
+            result.append(nums[n+i])
+        return result
+
+
+nums = [2, 5, 1, 3, 4, 7]
+n = 3
+print(Solution().shuffle(nums, n))
+nums = [1, 2, 3, 4, 4, 3, 2, 1]
+n = 4
+print(Solution().shuffle(nums, n))
+nums = [1, 1, 2, 2]
+n = 2
+print(Solution().shuffle(nums, n))