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

Link

코딩테스트 연습 - 음양 더하기

Solution

해당 문제는 zip 함수를 사용하면 간단하게 해결할 수 있다.

zip 함수의 간단한 예시를 살펴보고 넘어가도록 하겠다.

// zip 함수 예시 //

nums = [1, 2, 3]
alps = ['A', 'B', 'C']

for num, alp in zip(nums, alps):
    print(num, alp)

for pair in zip(nums, alps):
    print(pair)

...

<output>
1 A
2 B
3 C
(1, 'A')
(2, 'B')
(3, 'C')

def solution(absolutes, signs):
    answer = 0

    for absolute, sign in zip(absolutes, signs):
        if sign:
            answer += absolute
        else:
            answer -= absolute

    return answer

📌 Tip: zip()

Other Solution

def solution(absolutes, signs):
    return sum(absolute if sign else -absolute for absolute, sign in zip(absolutes, signs))