DFS를 활용했다.
answer는 target과 일치하면 1씩 누적하는 변수다. 두 함수에서 사용할 수 있도록 전역변수로 만들었다.
numbers와 길이, target, 0을 인자로 갖는 make_target을 호출한다.
i는 level을 측정하면서 numbers의 index로도 쓰인다. numbers의 끝까지 탐색을 했다면 numbers의 합이 target과 일치하는지 판단하고 일치한다면 answer에 1을 누적한다. 그리고 make_target 함수를 종료한다.
트리 레벨(i)에 따라 레벨에 있는 하나의 원소의 기호를 변경해준다.
numbers[i] *= 1
make_target(numbers, target, length, i+1)
numbers[i] *= -1
make_target(numbers, target, length, i+1)
answer = 0
def make_target(numbers, target, length, i):
global answer
if i == length:
if target == sum(numbers):
answer += 1
return
numbers[i] *= 1
make_target(numbers, target, length, i+1)
numbers[i] *= -1
make_target(numbers, target, length, i+1)
def solution(numbers, target):
global answer
length = len(numbers)
make_target(numbers, target, length, 0)
return answer
출처 https://programmers.co.kr/learn/courses/30/lessons/43165
'알고리즘 문제' 카테고리의 다른 글
프로그래머스 단어 변환 (0) | 2019.09.03 |
---|---|
프로그래머스 네트워크 (0) | 2019.09.02 |
프로그래머스 입국심사 (0) | 2019.09.01 |
프로그래머스 예산 (0) | 2019.08.31 |
프로그래머스 카펫 (0) | 2019.08.30 |