본문 바로가기

ETC

프로그래머스 DFS/BFS 타겟 넘버 python 풀이

반응형

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+1 = 3 +1+1-1+1+1 = 3 +1+1+1-1+1 = 3 +1+1+1+1-1 = 3 사용할 수 있는 숫자가 담긴 배열 numbers, 타겟 넘버 target이 매개변수로 주어질 때 숫자를 적절히 더하고 빼서 타겟 넘

programmers.co.kr

이 문제는... 다른 사람의 풀이 중에... 재귀적으로 멋지게 푼 풀이가 있어서 약간 수정해서 가져왔다...

리스트 맨 뒤에서 한개씩 target에서 빼며, 리스트 길이를 줄인다...

리스트를 전부 소진하였을 때... target = 0이면 1을 반환하는 식으로 간다.

재귀 호출을 사용하였으므로 DFS를 이용한 풀이이다.

def solution(numbers:list, target:int)->int:
    if not numbers: # numbers 리스트 전부 소진 
        result = (1 if target == 0 else 0) # target 맞으면 1 return 아니면 0 리턴
        return result 
    # numbers 리스트에 뭔가 남았을 경우 맨 뒤의 수 +,- 경우의 수 계산
    return solution(numbers[:-1],target-numbers[-1]) + solution(numbers[:-1],target+numbers[-1])

 

반응형