문제
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;
}
'PS > BaekJoon' 카테고리의 다른 글
[백준/c++] 벽 부수고 이동하기 2 (0) | 2022.03.14 |
---|---|
[백준/c++] 2206 벽 부수고 이동하기 (0) | 2022.03.14 |
[백준/c++] 16935 배열 돌리기 3 (0) | 2022.03.11 |
[백준/c++] 16927 배열 돌리기 2 (0) | 2022.03.10 |
[백준/c++] 16926 배열 돌리기 1 (0) | 2022.03.10 |