알고리즘 풀이/프로그래머스

[Level 2] 타겟 넘버

mhko411 2021. 9. 5. 15:25
728x90

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


접근

전달받은 numbers의 숫자들을 더하거나 빼서 target을 만들 수 있는 경우의 수가 몇 개인지 구해야한다. 따라서 DFS를 통해 모든 경우의 수를 먼저 구했다. 

 

구현

- DFS 내에서 현재 인덱스와 지금까지 합을 더하거나 뺄 때로 나눠서 재귀호출을 진행하였고

- idx가 N이 될 때는 모든 선택이 된 것이기 때문에 지금까지 더한 것이 target과 같을 때 

- 최종해를 더해주었다.

def solution(numbers, target):
    answer = 0
    N = len(numbers)

    def dfs(idx, total):
        nonlocal answer
        if idx == N:
            if total == target:
                answer += 1
            return
        dfs(idx+1, total + numbers[idx])
        dfs(idx+1, total - numbers[idx])

    dfs(1, -numbers[0])
    dfs(1, numbers[0])
    return answer

'알고리즘 풀이 > 프로그래머스' 카테고리의 다른 글

[Level 3] 디스크 컨트롤러  (0) 2021.09.06
[Level 2] 거리두기 확인하기  (0) 2021.09.05
[Level 2] 순위 검색  (0) 2021.09.05
[Level 2] 메뉴 리뉴얼  (0) 2021.09.02
[Level 2] 오픈채팅방  (0) 2021.08.30