[프로그래머스 LV.1] 음양 더하기

프로그래머스 LV.1 모음

음양 더하기

문제 설명

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

제한 사항

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

입출력 예

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

입출력 예 설명

입출력 예 #1

• signs가 [true, false, true] 이므로, 실제 수들의 값은 각각 4, -7, 12입니다.
• 따라서 세 수의 합인 9를 return 해야 합니다.

입출력 예 #2

• signs가 [false, false, true] 이므로, 실제 수들의 값은 각각 -1, -2, 3입니다.
• 따라서 세 수의 합인 0을 return 해야 합니다.

---

제출

import Foundation

func solution(_ absolutes:[Int], _ signs:[Bool]) -> Int {
    var result = 0

    for i in 0..<absolutes.count{
    	result = signs[i] ? result+absolutes[i] : result-absolutes[i]
    }

    return result
}

결괏값을 저장할 변수 result를 생성한다.
absolutes 배열의 크기만큼 for문을 실행한다.
signs 배열의 요소가 true 라면 result에 absolutes배열의 요소를 더하고 아니라면 absolutes 배열의 요소를 뺀다.

다른 풀이

import Foundation

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

map함수를 사용하여 signs의 요소가 ture라면 양수의 absolutes 배열을 아니라면 음수의 absolutes을 반환한다.
reduce함수를 사용하여 map 함수에서 반환한 요소들을 더한다.