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

+ Recent posts