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

📒  문제 정보

• 출처          : 월간 코드 챌린지 2 
• 난이도      : Level 1
• 문제 링크  : 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	signs	result
[4,7,12]	[true,false,true]	9
[1,2,3]	[false,false,true]	0

 

입출력 예 설명

입출력 예 #1

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

입출력 예 #2

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

 

 

🔑  문제 풀이

방법 1. 단순히 반복문 활용

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. zip 함수 활용

zip() 함수 및 sum(a if b else a') 함수 등을 활용하여 논리 그대로 훨씬 간단하고 직관적으로 푼 풀이 역시 첨부한다.

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

 

 

 

💡  What I learned

1. 저번 포스팅에서 알게 된 zip() 함수의 활용법
2. 함수 (sum()) 안에서의 조건문 및 반복문이 활용 가능하다는 점!!