[백준] 1182번 부분수열의 합 C++

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

1182번: 부분수열의 합

백트래킹을 통해서 주어진 수열의 부분수열의 합이 S가 되는 경우의 수를 구해야 한다.

{1,2,3,4,5}수열의 부분 수열에는 {1},{2},{1,3},{1,2,4,5} 등 이 있을 수 있다.

 

경우의 수를 중복되게 카운트 하지 않도록 idx 매개변수를 통해 idx+1부터 N-1까지 방문을 안한 idx를 방문하도록 하였다.

그리고 백트래킹을 위해 함수 호출전 visited[i]=1 과 함수 호출 후 visited[i]=0을 해주었다.

#include<iostream>
#include<algorithm>
#include<unordered_map>
#include<cstring>
#include<math.h>
#include<queue>
#include<vector>

using namespace std;

int Count = 0;
const int INF = 123456789;
int N, S;
int arr[20];
int visited[20];

void back(int idx, int sum) {

	if (sum == S){
		Count++;
	}

	if (sum == INF) {
		for (int i = 0; i < N; i++) {
			visited[i] = 1;
			back(i, arr[i]);
			visited[i] = 0;
		}
	}
	else {
		for (int i = idx + 1; i < N; i++) {
			if (visited[i] == 1)continue;
			visited[i] = 1;
			back(i, sum + arr[i]);
			visited[i] = 0;
		}
	}

}




int main(void)
{
	ios::sync_with_stdio(false); cin.tie(NULL); cout.tie(NULL);

	cin >> N >> S;

	
	for (int i = 0; i < N; i++)cin >> arr[i];

	back(0, INF);

	cout << Count;

	
}