Leetcode 1470 풀이(파이썬, 스위프트)

i번째 인덱스를 append 시키고 n+i번째 인덱스를 append시키는 식으로 셔플해주면 된다.

 

파이썬:

class Solution:
    def shuffle(self, nums: List[int], n: int) -> List[int]:
        
        arr=[]
        for i in range(n):
            arr.append(nums[i])
            arr.append(nums[n+i])
        return arr

스위프트:

class Solution {
    func shuffle(_ nums: [Int], _ n: Int) -> [Int] {
        
        var a = [Int]()
        for i in 0..<n{
            a.append(nums[i])
            a.append(nums[n+i])
        }
        return a
    }
}