음양 더하기

nosungmin 2023. 1. 27. 11:11

문제 설명

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

제한사항

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

입출력 예absolutessignsresult

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

이런 문제는 처음봐서 어리둥절하였다.

내가 풀어낸 코드

def solution(absolutes, signs):
    answer=0
    for i in range(len(absolutes)):
        if signs[i]==True:
            answer += absolutes[i]
        elif signs[i]==False:
            answer -= absolutes[i]
    return answer

하지만 더 좋은 코드가 있었다.

너무나 간단한 코......이게 어떻게 돌아가는 걸까?

def solution(absolutes, signs):

return sum([absolutes[i] if signs[i] else -absolutes[i] for i in range(len(absolutes))])

위 코드는 파이썬 리스트 내포(list comprehension)를 사용하여 풀어냈다고 한다.

위에서 사용한 리스트 내포의 대해서 알아보자

이 코드는 리스트 내포(List comprehension)를 사용해 absolutes 리스트와 signs 리스트를 이용해 실제 정수들의 합을 구하는 코드입니다.

우선, for i in range(len(absolutes))로 absolutes 리스트의 길이만큼 반복문을 돌립니다. 그리고, if signs[i] else -absolutes[i] 부분은 signs[i] 가 참일 경우 absolutes[i]를 양수로, 거짓일 경우 -absolutes[i]를 음수로 인식하는 코드입니다. 그리고 이러한 값들을 리스트로 저장하는 구문이 [absolutes[i] if signs[i] else -absolutes[i] for i in range(len(absolutes))] 이고, 이 리스트를 sum 함수를 사용해 합계를 구하는 것 입니다.

라고 한다.

기억해두자