프로그래머스 Lv.1 Swift 음양더하기

 

 

문제

어떤 정수들의 절댓값을 차례대로 담은 정수 배열 absolutes와 이 정수들의 부호를 차례대로 담은 불리언 배열 signs가 매개변수로 주어진다. 실제 정수들의 합을 구하여 return 하도록 함수를 완성하라

 

 

제한사항

• absolutes의 길이는 1이상 1000이하 (absolutes의 모든 수는 각각 1이상 1000이하)
• signs의 길이는 absolutes의 길이와 같음. (signs[i]가 참이면 absolutes[i]의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미한다.)

 

 

입출력 예

absolutes	signs	result
[4,7,12]	[true,false,true]	9
[1,2,3]	[false,false,true]	0

 

 

풀이

    func solution(_ absolutes:[Int], _ signs:[Bool]) -> Int {
        var answer = 0
        for i in 0..<absolutes.count {
            if signs[i] == false {
                answer -= absolutes[i]
            } else {
                answer += absolutes[i]
            }
        }
        return answer
    }

 

 

축약한 풀이

 

zip과 reduce를 활용하여 람다식으로 구현하는 방법도 있어서 코드를 한결 더 효율적이고 깔끔하게 만들 수 있다.

zip 만들어서 하는 걸 차마 생각 못했다.. ㄱ- rx에서는 잘 썼건만..

 

 func solution(_ absolutes:[Int], _ signs:[Bool]) -> Int {
    return zip(absolutes, signs)
        .map { $1 ? $0 : -$0 }
        .reduce(0, +)
}

 

또 다른 방법

func solution(_ absolutes:[Int], _ signs:[Bool]) -> Int {
    var answer = 0
    answer = (0..<absolutes.count).map { signs[$0] ? absolutes[$0] : -absolutes[$0]}.reduce(0, +)
    return answer
}