Solution 1 - Simulation (Extra Space)

Given an array nums of 2n elements in the form [x1, x2, ..., xn, y1, y2, ..., yn], return the array in the form [x1, y1, x2, y2, ..., xn, yn]. We iterate through indices 0 to n-1, picking one element from each half and appending them to a result list.

TimeComplexity: O(n)

SpaceComplexity: O(n)

""" 1470.1 - Solution 1 - Simulation (Extra Space) """

#####################################################################################
# Imports
#####################################################################################
from typing import List

#####################################################################################
# Classes
#####################################################################################
class Solution:
    """Solution Class"""

    def shuffle(self, nums: List[int], n: int) -> List[int]:
        """Shuffle the Array Function"""
        result = []
        for i in range(n):
            result.append(nums[i])
            result.append(nums[i + n])
        return result


#####################################################################################
# Functions
#####################################################################################
def testcase():
    """Test Function"""
    print(Solution().shuffle([2, 5, 1, 3, 4, 7], 3))
    print(Solution().shuffle([1, 2, 3, 4, 4, 3, 2, 1], 4))
    print(Solution().shuffle([1, 1, 2, 2], 2))


#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
    testcase()

Solution 2 - Using `zip()` and List Comprehension

Same approach but leverages Python's zip() to pair elements from the first and second halves, then flattens the pairs into a single list using a list comprehension.

TimeComplexity: O(n)

SpaceComplexity: O(n)

""" 1470.2 - Solution 2 - Using zip() and List Comprehension """

#####################################################################################
# Imports
#####################################################################################
from typing import List

#####################################################################################
# Classes
#####################################################################################
class Solution:
    """Solution Class"""

    def shuffle(self, nums: List[int], n: int) -> List[int]:
        """Shuffle the Array Function"""
        return [val for pair in zip(nums[:n], nums[n:]) for val in pair]


#####################################################################################
# Functions
#####################################################################################
def testcase():
    """Test Function"""
    print(Solution().shuffle([2, 5, 1, 3, 4, 7], 3))
    print(Solution().shuffle([1, 2, 3, 4, 4, 3, 2, 1], 4))
    print(Solution().shuffle([1, 1, 2, 2], 2))


#####################################################################################
# Main
#####################################################################################
if __name__ == "__main__":
    testcase()

1470 - Shuffle the Array

Solution 1 - Simulation (Extra Space)

Solution 2 - Using zip() and List Comprehension

Solution 2 - Using `zip()` and List Comprehension