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

문제

 

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

 

입력

 

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

 

출력

 

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

 

코드

 

#include <iostream>
using namespace std;
int a[21];

int main()
{
	int n, s;
	cin >> n >> s;
	for (int i = 0; i < n; i++) {
		cin >> a[i];
	}

	int ans = 0;
	for (int i = 1; i < (1 << n); i++) {
		int sum = 0;
		for (int j = 0; j < n; j++) {
			if (i&(1 << j)) {
				sum += a[j];
			}
		}
		if (sum == s) {
			ans += 1;
		}
	}

	cout << ans << '\n';


	return 0;
}

 

메모

 

*Bit mask 0 ~ (N-1)

1. i 추가  s | (1<<i)

1. i 검사  s & (1<<i)

1. i 제거  s &~(1<<i)

1. i 토글  s ^ (1<<i)