본문 바로가기
코딩테스트

[파이썬] 음양 더하기 - 월간 코드 챌린지 시즌2 [CODING TEST #12]

by ALTERww 2022. 7. 21.
320x100

https://school.programmers.co.kr/learn/courses/30/lessons/76501

 

 

무지성으로 아래 코드처럼 짜기는 했다.

 

def solution(absolutes, signs):
    answer = 0
    for i in range(len(absolutes)):
        if signs[i] == True : t = 1
        else : t = -1
        absolutes[i] *= t
        answer +=  absolutes[i]
    return answer

 

더 줄일 수 있을 것 같아서 좀 더 고민해보니 zip을 이용해 한 줄로 짤 수 있었다.

 

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

 

댓글