[프로그래머스] 음양 더하기(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] 의 실제 정수가 양수임을, 그렇지 않으면 음수임을 의미합니다.

---

 

✔️ 입출력 예

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 해야 합니다.

---

🗒 풀이

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

    return answer

zip 함수를 사용해 같이 for문을 돌린다.

만약, 부호가 True라면 answer에 양을 더해주고, False라면 answer에 음을 더해준다.

---

🤔 후기

앞서 내적에서 응용했던 zip함수를 사용했다. 때문에 비교적 쉽게 풀 수 있었다.

숏코딩을 할 수도 있었겠지만, 조건문이 있는 만큼 좀 더 코드를 명확하게 짜고 싶어서 사용하지 않았다. 물론 조건문이 까다로운게 아니므로 이제와 생각해보면 상관없었을 것 같다.

한줄로 쓰고 싶다면 다음과 같이 코드를 짤 수 있겠다.

sum(absolutes if sign else -absolutes for absolutes, sign in zip(absolutes, signs))