알고리즘 문제
[백준] 1182번 부분수열의 합
feelcoding
2020. 2. 20. 03:39
728x90
https://www.acmicpc.net/problem/1182
1182번: 부분수열의 합
첫째 줄에 정수의 개수를 나타내는 N과 정수 S가 주어진다. (1 ≤ N ≤ 20, |S| ≤ 1,000,000) 둘째 줄에 N개의 정수가 빈 칸을 사이에 두고 주어진다. 주어지는 정수의 절댓값은 100,000을 넘지 않는다.
www.acmicpc.net
#include <iostream>
#include <vector>
#include <tuple>
using namespace std;
int n;
int s;
vector<int> v;
int cnt = 0;
void combination(vector<bool> visited, vector<pair<int, int>> result, int num, int k) {
if (num == k) {
int total = 0;
for (int i = 0; i < result.size(); i++) {
total += result[i].first;
}
if (total == s) cnt++;
return;
}
for (int i = 0; i < n; i++) {
if (num == 0) {
if (!visited[i]) {
visited[i] = true;
result[num].first = v[i];
result[num].second = i;
combination(visited, result, num + 1, k);
visited[i] = false;
}
}
else {
if (result[num - 1].second < i && !visited[i]) {
visited[i] = true;
result[num].first = v[i];
result[num].second = i;
combination(visited, result, num + 1, k);
visited[i] = false;
}
}
}
}
int main() {
cin.tie(NULL);
ios_base::sync_with_stdio(false);
cin >> n >> s;
v = vector<int>(n);
for (int i = 0; i < n; i++) {
cin >> v[i];
}
for (int i = 1; i <= n; i++) {
vector<bool> visited(n);
vector<pair<int, int>> result(i);
combination(visited, result, 0, i);
}
cout << cnt;
return 0;
}
pair의 second에는 인덱스를 넣어주었다.
728x90