<백준> 1182

진기명기·2025년 9월 19일

코딩테스트<C++>

목록 보기
145/212

부분수열의 합

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

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

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

백트래킹으로 start의 위치를 옮겨가며 s와 같은 합이 나올때마다 count를 체크해주는 전형적인 백트래킹 유형의 문제이다.

int n, s;
vector<int> v;
int arr[20];
bool visited[20];
int cnt;
int sum;

void DFS(int idx, int start,int sum)
{
	if (sum == s && idx > 0)
		cnt++;

	for (int i = start; i < n; i++)
	{
		if (!visited[i])
		{
			visited[i] = true;
			arr[idx] = v[i];
			DFS(idx + 1, i + 1, sum+v[i]);
			visited[i] = false;
		}
	}
}

int main() {
	ios::sync_with_stdio(false);
	cin.tie(NULL);
	cout.tie(NULL);

	cin >> n >> s;

	for (int i = 0; i < n; i++)
	{
		int value;
		cin >> value;
		v.push_back(value);
	}

	DFS(0,0,0);

	cout << cnt;
}

0개의 댓글