[Programmers/프로그래머스]음양 더하기

문제 설명

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

제한사항

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

입출력 예

absolutes	signs	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 해야 합니다.

풀이

이 문제는 zip(), sum(), 리스트 컨프리헨션을 통해 쉽게 구할 수 있다.

• zip()으로 absolutes와 signs를 쌍으로 묶어서 for문에 전달한다.
• 리스트 컴프리헨션에서 조건문을 통해 signs[index]의 값이 참이면 absolutes[index]는 그대로, 거짓이면 음수로 리스트를 생성한다.

이렇게 만들어진 리스트를 sum()로 전부 더한 값을 리턴하면 된다.

 

def solution(absolutes, signs):
    return sum([i if op else -i for i,op in zip(absolutes,signs)])