[Swift] 프로그래머스 - 배열 조각하기

 

매일 Python 으로 문제 풀다가 Swift로 시작하니까 정말 어려운 것 같네요.

이번 문제는 for문 순회를 통해 배열을 슬라이싱 하는 문제입니다.

 

import Foundation

func solution(_ arr:[Int], _ query:[Int]) -> [Int] {
    var first : Int = 0
    var last : Int = arr.count
    
    for index in 0..<query.count {
        if index % 2 == 0 {
            last = first + query[index]
            // 짝수일 때 마지막 인덱스 저장
        } else {
            first += query[index]
            // 홀수일 때 마지막 인덱스 저장
        }   
    }

    return Array(arr[first...last])
    // 슬라이싱한 배열을 그대로 제출하면 ArraySlice 타입이 되기 때문에 에러가 발생
    // Array 타입으로 바꾼 다음에 Return
}