[프로그래머스] 타겟넘버 - 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 이하인 자연수입니다.

풀이

숫자의 개수가 작으므로 백트래킹으로 풀 수 있다.

각 선택지를 일단 해보고 target 넘버를 만들 수 있으면 1 아니면 0을 리턴한다.

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

이런 식으로 재귀하게 된다.

 

 

스택을 이용해서 푸는 방법

• results에 전 결과를 저장한다.
• stack에 전 결과에 더하거나 뺀 결과를 저장한다.
• results를 업데이트한다.
• results 안에 target의 갯수를 센다.

def solution(numbers, target):
    answer = 0
    results = [0] # 전 시행의 결과 저장
    
    for number in numbers:
        stack = [] # 각 시행의 모든 선택지 저장
        
        for _ in range(len(results)):
            pop = results.pop()
            stack.append(pop + number)
            stack.append(pop - number)
        results = stack
        print(results)
        
    for result in results:
        if result == target:
            answer += 1
            
    return answer