본문으로 바로가기

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

ans = 0
def solution(numbers, target):
    dfs(0, numbers, target, 0)
    return ans
    
def dfs(s, num, tgt, depth):
    global ans
    if depth == len(num):
        if s == tgt:
            ans += 1
        return
    dfs(s+num[depth], num, tgt, depth+1)
    dfs(s-num[depth], num, tgt, depth+1)

일주일간 플젝 하느라 떨어진 감을 찾기 위해 다시 쉬운것부터 풀어보는 중이다.

DFS로 풀었고, 재귀함수를 이용했다.

from collections import deque

def solution(numbers, target):
    answer = 0
    q = deque([(0, 0)])
    while q:
        sum, i = q.popleft()
        if i == len(numbers):
            if sum == target:
                answer += 1
        else:
            number = numbers[i]
            q.append((sum + number, i + 1))
            q.append((sum - number, i + 1))

    return answer

이렇게 BFS로도 해결할 수 있다.