본문 바로가기

PS/BaekJoon

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

문제

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

입력

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

출력

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

구현해야할 조건

부분 수열이니까 수열을 뽑아서 문제를 해결하면 됩니다. 조합으로다가 prev_permutation을 사용했는데 이 방법을 사용해도 괜찮고, next_permutation으로 사용해도 상관없습니다!

code

#include<iostream>
#include<vector>
#include<algorithm>

#define MAX 21
using namespace std;

int n,s;
int arr[MAX];

int main()
{
	cin >> n >> s;
	
	for(int i = 0; i < n; i++)
		cin >> arr[i];
		
	long long cnt = 0;
	
	for(int r = 1; r <= n; r++)
	{
		vector<bool> comb(n,false);
		
		for(int j = 0; j < r; j++) // true에 넣고! 그 값을 넣은 애들만!
			comb[j] = true;
			
		do{
			
			long long sum = 0;
			
			for(int i = 0; i < n; i++)
			{
				if(!comb[i])
					continue;
				else
				{
					sum += arr[i];
				}
			}
			
			if(sum == s)
				cnt++;

		}while(prev_permutation(comb.begin(),comb.end()));
		
	}
	
	cout << cnt;
	
	return 0;
}