[백준 1182 - C++] 부분 수열의 합 : 브루트포스

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

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

풀이

이 문제는 수열을 순서대로 따라가는 재귀를 이용한다.
이때 합에 현재 idx번째 수를 더하거나 안더하거나 이 두가지 경우의 수가 있다.
따라서 더하는 경우의 함수도 호출하고 더하지 않는 경우의 함수도 호출한다.
이때 아무것도 안 더해서 0이 되는 경우를 세는 일을 방지해야하므로 add_cnt ==0 이고 cur_num == s인 경우만 cnt++해준다.
그리고 cnt++을 한 경우 그 이후에 아무것도 안더해도 그 경우를 count하게 되지 않도록 add_cnt = 0으로 다시 만들어준다.
또한 cur_idx와 N이 같은 경우는 모든 수열을 다 보았다는 의미이므로 return한다.

#include <iostream>
#include <cstdio>

int arr[21];
int cnt;
int n, s;

void adding (int cur_num, int cur_idx, int add_cnt){
    if(cur_num==s && add_cnt!=0){
        cnt++;
        add_cnt=0;
    }
    cur_idx++;
    if(cur_idx == n)
        return;
    adding(cur_num, cur_idx, add_cnt); //현재 idx의 수 안 더함
    adding(cur_num+arr[cur_idx], cur_idx, add_cnt+1); //현재 idx의 수 더함
    
}

int main (void){
    
    scanf("%d %d", &n, &s);
    
    for(int i=0; i<n; i++)
        scanf("%d", &arr[i]);
    
    adding(0, 0, 0);    //첫번째 수 안 더함
    
    adding(arr[0], 0, 1);   //첫번째 수 더함
    
    printf("%d\n", cnt);
}