[Programmers] 등차수열의 특정한 항만 더하기

프로그래머스

 

내 풀이

import Foundation

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

 

 

다른 사람의 풀이

import Foundation

func solution(_ a: Int, _ d: Int, _ included: [Bool]) -> Int {
    return included.indices.filter { included[$0] }.map { a + d * $0 }.reduce(0, +)
}