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

시작하며...

모든 알고리즘 문제 풀이는 제가 직접 짜서 정답을 맞춘 결과만을 공유합니다.

마지막 'More Solution'은 다른 정답자들 풀이 중 생각지 못했던 부분들이나 좋게 느껴진 풀이법 몇개를 가져와서 공유하였습니다.

 

[Level 1] 음양 더하기

프로그래머스

 

문제 설명

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

---

제한사항

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

입출력 예

입출력 예 설명

입출력 예 #1

• signs가 [true,false,true] 이므로, 실제 수들의 값은 각각 4, -7, 12입니다.
• 따라서 세 수의 합인 9를 return 해야 합니다.

입출력 예 #2

• signs가 [false,false,true] 이므로, 실제 수들의 값은 각각 -1, -2, 3입니다.
• 따라서 세 수의 합인 0을 return 해야 합니다.

 

My Solution

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

signs에 따라 양수인지 음수인지 정해지는 문제이다. 

• 주어진 배열 absolutes와 signs를 zip 함수로 묶어서 각 원소들을 이용한다
• sign이 True이면 absolute가 양수라는 의미이므로 answer에 더해주고
• sign이 False이면 absolute가 음수이므로 answer에서 빼주면 된다.

 

More Solution

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

위 풀이를 sum 함수와 삼항연산을 이용해 한줄로 풀이한 문제를 가져와 보았다.