본문 바로가기

Algorithm/백준

[백준] 1182번: 부분수열의 합

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

 

1182번: 부분수열의 합

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

www.acmicpc.net

전에 재귀로 풀었던 문제인데 가볍게 비트마스킹 연습할 겸 다시 풀어보았다.

근데 이 문제는 재귀로 푸는게 더 빠르네..ㅋㅋㅋㅋㅋㅋㅋㅋ

 

#include <iostream>
using namespace std;

int n, s;
int arr[20];
int cnt = 0;

void calc(int i, int sum) {
	if (i == n) {
		if (sum == s) cnt++;
	}
	else {
		calc(i + 1, sum);  //arr[i] 안 더하는 경우
		calc(i + 1, sum + arr[i]);  //arr[i] 더하는 경우
	}
}

int main() {

	cin >> n >> s;
	for (int i = 0; i < n; i++) {
		cin >> arr[i];
	}

	calc(0, 0);

	if (s == 0) cnt--;

	cout << cnt;
}

이건 재귀로 푼거

 

 

#include <iostream>

using namespace std;

int main() {
	int n, s;
	int arr[20];
	cin >> n >> s;
	for (int i = 0; i < n; i++) cin >> arr[i];

	int cnt = 0;

	for (int i = 1; i < (1 << n); i++) {
		int sum = 0;
		for (int j = 0; j < n; j++) {
			if (i & (1 << j)) {
				sum += arr[j];
			}
		}
		if (sum == s) cnt++;
	}

	cout << cnt;
}

 

비트마스킹으로 푼거

'Algorithm > 백준' 카테고리의 다른 글

[백준] 11724번: 연결 요소의 개수  (0) 2020.04.08
[백준] 1012번: 유기농 배추  (0) 2020.04.06
[백준] 1874번: 스택 수열  (0) 2020.03.23
[백준] 5076번: Web Pages  (0) 2020.03.22
[백준] 2304번: 창고 다각형  (0) 2020.03.22