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

부분 수열의 합

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

풀이

비트마스크로 주어진 N개의 정수를 뽑는 방법을 정하고 그로부터 나오는 정수들을 더해서 S가 되는지 확인해 보는 방법이다.

#include <iostream>
using namespace std;

int arr[20];

int fullySearch(int N, int S)
{
	int count = 0;
	for(int i = 1; i < (1 << N); ++i) {
		int sum = 0;
		for(int j = 0; j < N; ++j)
			if(i & (1 << j))
				sum += arr[j];
		if(sum == S)
			++count;
	}

	return count;
}

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

	cout << fullySearch(N, S);
	return 0;
}

스터디원 중 한 명이 조금 다른 방법으로 풀었는데 그 실행속도가 훨씬 빨랐다. 재귀였다. 모든 비트마스크는 재귀로 고칠 수 있다고 했다.
이 문제의 경우 0부터 1111~111(20자리)를 차례로 돌면서 재귀호출로 sum값을 넘겨주는 방식이었는데, 비트마스크가 정해지면 for루프로 sum을 구하는 내 방식보다 빠르다.

끝