Level 1. 음양 더하기

04. 음양 더하기

코딩테스트 연습 > 월간 코드 챌린지 시즌2 > 음양 더하기
https://programmers.co.kr/learn/courses/30/lessons/76501

문제 설명

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

제한 조건

ㆍ absolutes의 길이는 1 이상 1,000 이하입니다.

ㆍ absolutes의 모든 수는 각각 1 이상 1,000 이하입니다.

ㆍ signs의 길이는 absolutes의 길이와 같습니다.

ㆍ signs(i) 가 참이면 absolutes(i) 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

입출력 예

absolutes = (4,7,12) signs = (true, false, true) result = 9

구상

1. zip(a,b)를 사용하여 b가 True일 경우 +, False일 경우 -를 적용하여 result 에 값을 추가한다.

풀이 방법

예전 풀이

def solution(absolutes, sings):
    answer = 0
    plus_minus = []    
    for i in sings:
        if i == True:
            plus_minus.append(1)
        else:
            plus_minus.append(-1)

    for i in range(len(absolutes)):
        answer += plus_minus[i]*absolutes[i]    
    return answer

zip을 이용한 풀이

def solution(absolutes,signs):
    result = 0
    for a,s in zip(absolutes,signs):
        if s == True:
            result+=a
        else:
            result-=a
    return result

한줄 코딩

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