문제 url:
난이도: Lv.2
권장 시간: 20분
권장 시간 복잡도: O(2^N)
문제 설명
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
이 매개변수로 주어질 때,
숫자를 적절히 더하고 빼서 타겟 넘버를 만드는 방법의 수를 반환하는 solution
함수를 작성하세요.
제한사항
- 주어지는 숫자의 개수는 2개 이상 20개 이하입니다.
- 각 숫자는 1 이상 50 이하인 자연수입니다.
- 타겟 넘버는 1 이상 1000 이하인 자연수입니다.
입출력 예
numbers | target | return |
---|---|---|
[1, 1, 1, 1, 1] |
3 | 5 |
[4, 1, 2, 1] |
4 | 2 |
입출력 예 설명
예제 #1
문제 예시와 같습니다.
예제 #2
+4+1-2+1 = 4
+4-1+2-1 = 4
총 2가지 방법이 있으므로, 2
를 반환합니다.
내 풀이
def solution(numbers, target):
answer = 0
cnt =0
dx = [-1,1]
def dfs(numlist,start):
nonlocal cnt
if start == len(numlist):
if sum(numlist) == target:
cnt+=1
return sum(numlist)
for i in range(2):
temp = numlist[start]
numlist[start] *= dx[i]
dfs(numlist,start+1)
numlist[start] = temp
dfs(numbers,0)
return cnt
'Algorithms > DFS' 카테고리의 다른 글
[Programmers] 여행경로 (1) | 2024.12.30 |
---|---|
[Programmers] 네트워크 - 그래프 (0) | 2024.12.16 |