본문 바로가기
성장중/알고리즘 인터뷰

[Programmers] 음양 더하기

반응형

https://programmers.co.kr/learn/courses/30/lessons/76501

 

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

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

programmers.co.kr

문제 설명

어떤 정수들이 있습니다. 이 정수들의 절댓값을 차례대로 담은 정수 배열 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 i, j in zip(absolutes, signs):
        if j == True:
            answer += i
        else: 
            answer -= i
    return answer

저번에 문제를 풀면서 zip함수를 사용에 아주 조금 익숙해져서 사용했다.
 i = absolutes, j = signs 들어가 있다.
j가 True면 i를 answer에 더해주고, False라면 빼주기.


다른 해답


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

한 줄로 표현했지만, 많이 어려운 문제가 아니라 똑같은 거 같다.

 

반응형

'성장중 > 알고리즘 인터뷰' 카테고리의 다른 글

[Programmers]3진법 뒤집기  (0) 2021.07.16
[Programmers] 약수의 개수와 덧셈  (0) 2021.07.10
[Programmers] 폰켓몬  (0) 2021.07.08
[Programmers] 모의고사  (0) 2021.07.06
[Programmers]내적  (0) 2021.07.05