알고리즘

programmers 타겟 넘버(Python)

jenyy 2021. 6. 6. 18:53

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

 

코딩테스트 연습 - 타겟 넘버

n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다. -1+1+1+1+1 = 3 +1-1+1+1+

programmers.co.kr

def dfs(count, cnt, numbers, target):
    global answer
    if count == len(numbers):
        if cnt == target:
            answer += 1
        return
    
    dfs(count + 1, cnt + numbers[count], numbers, target)
    dfs(count + 1, cnt - numbers[count], numbers, target)
    
answer = 0
def solution(numbers, target):
    dfs(0, 0, numbers, target)
    return answer