출처: Programmers [월간 코드 챌린지 시즌2]음양 더하기
문제 설명:
어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 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 해야 합니다.
보완해야 할 점:
- enumerate함수를 썼지만, zip()함수를 써서 사용하는 방법도 고려해볼 만 하다
코드설명:
- enumerate()함수를 이용해 absolute 리스트와 같은 인덱스를 가지는 sign 리스트 안 요소를 구함
- true시 answer에 absolute 요소 더하기, false시 빼기 실행
코드:
1
2
3
4
5
6
7
8
9
def solution(absolutes, signs):
answer = 0
for i, number in enumerate(absolutes):
if signs[i]:
answer += number
else:
answer -= number
return answer
우수답변자 코드:
1
2
3
def solution(absolutes, signs):
return sum(absolutes if sign else -absolutes for absolutes, sign in zip(absolutes, signs))