본문 바로가기
[업무 지식]/Algorithm

[list comprehension] 음양 더하기

by 에디터 윤슬 2024. 12. 19.

링크

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

 

프로그래머스

SW개발자를 위한 평가, 교육, 채용까지 Total Solution을 제공하는 개발자 성장을 위한 베이스캠프

programmers.co.kr

 

문제

 

정답 1

def solution(absolutes, signs):
    answer = []
    for val, bul in zip(absolutes, signs):
        if bul == True:
            val = val
        else:
            val = -val
        answer.append(val)
    return sum(answer)

 

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

정답 2

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