알고리즘 문제 풀이/프로그래머스
[프로그래머스] 타겟 넘버
weekyear
2020. 12. 14. 00:08
코딩테스트 연습 - 타겟 넘버
n개의 음이 아닌 정수가 있습니다. 이 수를 적절히 더하거나 빼서 타겟 넘버를 만들려고 합니다. 예를 들어 [1, 1, 1, 1, 1]로 숫자 3을 만들려면 다음 다섯 방법을 쓸 수 있습니다. -1+1+1+1+1 = 3 +1-1+1+1+
programmers.co.kr
<풀이>
아주 단순하게 재귀함수로 DFS를 구현하였다.
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
|
def solution(numbers, target):
len_n = len(numbers)
def dfs(idx, result):
if idx == len_n:
if result == target:
return 1
else:
return 0
return dfs(idx + 1, result + numbers[idx]) + dfs(idx + 1, result - numbers[idx])
answer = dfs(0, 0)
return answer
|
cs |
반응형