[프로그래머스] 음양 더하기

2021. 5. 20. 22:00파이썬

문제 설명

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

제한 사항

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

입출력 예

[4,7,12] [true,false,true] 9
[1,2,3] [false,false,true] 0

풀이

이번 문제는 엄청 간단하게 풀었다

우선 양수인지 음수인지 판별해 넣어줄 answer 배열을 만들어준 후, absolutes 길이만큼 반복문을 돌린 조건문으로 boolean배열인 signs가 참이면 absolutes[i]번째값을 넣었고, 거짓일 경우에는 음수로 바꿔서 넣었다.

그리고 sum함수를 통해 answer 배열 안에있는 모든 값을 더한 후 리턴해준다

def solution(absolutes, signs):
    answer = []
    for i in range(len(absolutes)):
        if signs[i] == True:
            answer.append(absolutes[i])
        else:
            answer.append(-absolutes[i])
    return sum(answer)