code/[프로그래머스] python

[코딩 테스트 연습] 음양 더하기

오리는짹짹 2023. 1. 3. 17:33

문제

어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 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 해야 합니다.

 

✔풀이

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

📝설명

문제에서 주어진 정수 배열 absolutes과 부호를 담은 signs의 순서를 맞추어 음양을 구분하여 더한 후 결과를 도출해야 한다.

signs에서 sign이 true라면 양수, false라면 음수이다.

absolutes의 len만큼 for문을 돌리면 인덱스 번호 순서(0, 1, 2 ...)대로 결과가 숫자를 사용할 수 있는데, 이를 index_num이라고 알기 쉽게 정의했다.

인덱스 번호가 나오게끔 for문을 만든 이유는, signs에서도 똑같은 인덱스 번호로 비교해야하기 때문이다.

for문 안에 signs 안이 True라면, absolutes의 숫자를 그대로 더하게 했고, false라면 answer에서 해당 수(absolutes(index_num)를 빼도록 했다.

 

👑모범 답안

🔎zip은 같은 길이의 리스트를 같은 인덱스끼리 잘라서 튜플로 묶어주는 함수이다.

 

👀느낀 점

: zip에 더 익숙해진다면, 언젠간 요긴하게 사용할 수 있을 것 같다.

 

💻 문제 출처

코딩테스트 연습 - 음양 더하기 | 프로그래머스 스쿨 (programmers.co.kr)