22. 등차수열의 특정한 항만 더하기

나의 코드

import Foundation

func solution(_ a:Int, _ d:Int, _ included:[Bool]) -> Int {
    
    var arr : [Int] = []
    var answer : Int = 0
    
    for i in 0 ... included.count-1 {
        
        arr.append(a + i*d)
        
    }
    
    for i in 0 ... included.count-1 {
        if included[i] == true {
            answer += arr[i]
        }
    }

    return answer
}

이렇게 생각한 이유. 먼저 Int타입을 가지는 배열 arr을 생성하여, 해당 배열에 등차수열의 값을 집어넣었다.

어차피 included의 길이 만큼 arr이 생성되기에, arr.count와 included.count가 같고 그에 해당하는 index 또한 같기때문에 또다른 for문을 생성하여 included를 중심으로 true일때 더해서 해결을 하려고하였다.