[c++] 부분수열의 합(1182), 백트래킹, dfs, 재귀

문제 https://www.acmicpc.net/problem/1182

문제

N개의 정수로 이루어진 수열이 있을 때, 크기가 양수인 부분수열 중에서 그 수열의 원소를 다 더한 값이 S가 되는 경우의 수를 구하는 프로그램을 작성하시오.

입력

첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.

출력

첫째 줄에 합이 S가 되는 부분수열의 개수를 출력한다.

예제 입력 1 복사

5 0
-7 -3 -2 5 8

예제 출력 1 복사

1

 

풀이

재귀적으로 DFS 수행, 트리 한쪽의 끝까지 도달한 후 return하면서 수행한다.
- check[] 배열로 현재 트리의 상태를 체크한다.
- 트리의 가장 깊은 곳까지 도달시 return한다
- 현재 상태의 트리의 sum() == S 맞는지 확인하여 count.
- 현재 상태 i의 check[i]=0으로 만들어주고 return.
- 이전 트리의 상태에서 또 다시 sum() == S 여부 확인 및 count.
- 모든 트리를 탐색한다 (완전탐색).

 

코드

 

알게 된 점

• 재귀적 DFS에 대해서 더욱 익숙해짐
• 언제 반환할지, 반환시에 트리의 상태를 생각해야함
• 예시를 트리로 그려가면서 어떻게 코드를 짤지 구상하면 도움이 된다.

 

git 코드

https://github.com/intlabSeJun/c-plus-coding-test/blob/master/10.%20%EB%B0%B1%ED%8A%B8%EB%9E%98%ED%82%B9/%EB%B6%80%EB%B6%84%EC%88%98%EC%97%B4%EC%9D%98%20%ED%95%A9(1182).cpp