프로그래머스 - '음양 더하기' (PYTHON 풀이)

문제:

 

숫자가 담긴 배열과, 부호가 담긴 배열을 입력받고, 각 부호를 반영해서 최종 숫자가 담긴 배열의 합을 리턴하는 문제이다.

 

풀이:

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

총합을 리턴할 total 변수를 0으로 먼저 선언하고, 반복문을 돈다.

필자는 부호를 담은 리스트인 signs를 전부 돌면서 signs의 값이 True일 경우, total에 숫자를 담은 리스트인 absolutes의 동일 인덱스 값을 더하고, False라면 음수이므로, absolutes[i]에 -를 붙여서 제외하도록 하였다. 'total -=' 로 해도 동일하게 동작할 것이다.

이후 반복문을 전부돌면 total에는 총합이 저장되어 있으므로 해당 변수를 리턴하면 된다.