프로그래머스 '타겟 넘버' 문제풀이 - Henry's Algorithm

 

 

문제 설명

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 이하인 자연수입니다.

입출력 예

numbers    target    return

[1, 1, 1, 1, 1]

3

5

입출력 예 설명

문제에 나온 예와 같습니다.

 

 

문제 푸는 방법:

모든 숫자에 +,-를 다 해보는 수밖에 없다.

+1 +1 +1 (?) (?) 이렇게 앞에서 3이 된다고 해도, 뒤에 값이 0이 된다는 보장이 없기 때문이다.

그래서 각 위치에 +/-를 해주어야 한다.

예제에 있어서는 2**5 번의 성능이 발생한다.

 

평균적으로 2**len(numbers) 만큼의 연산을 해야하는 것이다.

 

일단 DFS 로 풀 수 있다.

각 자리수로 들어가면서 해당 값을 +/- 연산을 모두 수행하는 것이다.

그리고 마지막 idx일때, 해당 값이 target과 같다면 result 라는 리스트에 1을 append해주고, target과 다르다면 0을 append 해준다.

==> 나중에 sum(result) 하면 1의 개수를 알 수 있다.

 

아래의 코드에서는 DFS는 재귀로 구현하였다.

정답 코드:

def solution(numbers, target):
    result = []
    DFS(numbers,0,0,target,result)
    return sum(result)

def DFS(arr,idx,current,target,result):
    if idx == len(arr):
        if current == target:
            return result.append(1)
        else:
            return result.append(0)
    DFS(arr,idx+1,current+arr[idx],target,result)
    DFS(arr,idx+1,current-arr[idx],target,result)
    return