Study hard

(c++)백준 1182번: 부분수열의 합 본문

백준/브루트 포스

(c++)백준 1182번: 부분수열의 합

Nimgnoej 2020. 8. 10. 11:11

https://www.acmicpc.net/problem/1182

 

1182번: 부분수열의 합

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

www.acmicpc.net

 

[풀이]

원소가 1개 이상인 모든 조합에 대해 원소의 합을 구하여 그 값이 S가 되면 카운트하는 방법으로 풀었다.

※처음 조합 구하는 함수를 호출할 때는 원소를 아무것도 고르지 않은 경우이므로 조건식을 이용해 그냥 통과하게 해야한다.

 

#include <iostream>
#include <vector>
using namespace std;

int N, S;
int ans = 0;//경우의 수
vector<int>Num(20);//N개의 정수 입력
vector<int>Set;
bool check[20];

bool Check() {
	int L = Set.size();
	int sum = 0;
	for (int i = 0; i < L; i++) {
		sum += Set[i];
	}
	if (sum == S)
		return true;
	return false;
}

//조합구하기
void getCase(int idx, int cnt) {
	if (cnt >= 1 && Check()) {
		ans++;
	}
	for (int i = idx; i < N; i++) {
		if (check[i] == true)continue;
		check[i] = true;
		Set.push_back(Num[i]);
		getCase(i, cnt+1);
		check[i] = false;
		Set.pop_back();
	}
}

void solution() {
	cin >> N >> S;
	for (int i = 0; i < N; i++) {
		cin >> Num[i];
	}
	getCase(0, 0);
	cout << ans << '\n';
}

int main() {
	solution();
	return 0;
}