[python] 타겟 넘버

문제 설명

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

-1+1+1+1+1 = 3 
+1-1+1+1+1 = 3
+1+1-1+1+1 = 3
+1+1+1-1+1 = 3
+1+1+1+1-1 = 3

사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 return 하도록 solution 함수를 작성해주세요.

제한 사항

• 주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
• 각 숫자는 1 이상 50 이하인 자연수입니다.
• 타겟 넘버는 1 이상 1000 이하인 자연수입니다.

나의 풀이

def solution(numbers, target):
    result = 0
    
    def dfs(index, path, total ):
        nonlocal result
        if path == len(numbers) :
            if total == target:
                result +=1 
            return
        
        for j in [-1,1]:
            dfs( index+1 , path+1 , total + (j * numbers[index]) )
                
    dfs(0,0,0)
    
    return result

• 깊이 우선 탐색(DFS) 를 활용한 풀이를 진행했다.

• index를 하나씩 더하면서 재귀문을 반복시켰는데, 한번의 dfs 함수 안에 numbers[index] 값을 +로 한번, -로 한번 더해줌으로써 모든 경우의 수를 계산할 수 있었다.

• 그 후, 만약 + - 를 통해 더한 값이 target과 일치할 경우 result의 값을 1 더해줌으로서 target과 일치하는 결과의 수를 세었다.