[프로그래머스/Lv. 1] 월간 코드 챌린지 시즌2 - 음양 더하기 (Python)

🔎 문제 링크

https://school.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] 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

 

🔦 문제 풀이

해당 문제는 길이가 같은 두 배열을 탐색하면서, signs 배열에서 음수의 정보를 받아 실제 계산을 달리하는 문제입니다. 아래 코드와 같이 signs[i]의 조건이 다른 경우엔 조건문을 통해 계산을 달리하고, 해당 계산을 answer에 반영하는 방식으로 풀이할 수 있습니다.

 

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

 

위 방식으로 풀어도 문제는 없지만, 내장 함수를 활용하여 코드를 더욱 깔끔하게 표현하면 아래와 같습니다.

 

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

 

내장 함수 zip()은 반복 가능한(iterable) 객체를 받아 인덱스에 있는 요소들을 결합하여 새로운 순회 가능한 객체를 생성하는 함수입니다. 위 코드와 같이 내장 함수를 활용하여 반복 가능한 두 리스트를 합쳐 반복문으로 활용할 수 있습니다.