코딩테스트

[프로그래머스/파이썬] 음양 더하기

Chef.Yeon 2022. 9. 22. 00:21

문제 설명

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

 

제한사항

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

 

입출력 예

absolutes signs result
[4,7,2] [true,false,true] 9
[1,2,3] [false,false,true] 0

 

내 풀이

signs 요소가 true 이면 continue

false 이면 absolutes 요소 값을 String 으로 변환하여 - 를 붙이고 int 형으로 변환

def solution(absolutes, signs):
    for i in range(len(absolutes)):
        if signs[i] :
            continue
        else :
            absolutes[i] = int("-"+str(absolutes[i]))
    return sum(absolutes)

 

다른 사람 풀이

zip 을 통해 데이터를 엮어 각 객체를 담고 있는 원소를 튜플 형태로 차례대로 접근

sign 이 True 이면 absolutes 값, false 이면 -absoultes 값

정수에 앞에 바로 - 붙인다는 생각을 왜 못했지

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

 

728x90