본문 바로가기

알고리즘/BOJ

1182번

1182번 - 부분 집합의 합


1. 재귀함수를 이용해서 풀이


2. 비트마스크를 이용해서 풀이


for문에서는 0인 공집합을 빼고 모든 부분집합을 도는 조건문이다.


아래의 for문을 


 for (int subset = set; subset; subset = set & (subset - 1))


아래처럼 바꿔도 동일하다.


for(int subset=1;subset<(1<<N);subset++)


<정답 코드 - 재귀함수>


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
#include<iostream>
#include<algorithm>
using namespace std;
 
int N;
int S;
int D[21];
int ans;
void dfs(int n, int sum)
{
    
    if (n == N)
    {
        if (sum == S)
        {
            ans += 1;
        }
 
        return;
    }
 
    dfs(n + 1, sum + D[n]);
    dfs(n + 1, sum);
 
 
    return;
}
int main()
{
    cin >> N >> S;
 
    for (int i = 0; i < N; i++)
    {
        cin >> D[i];
    }
    dfs(00);
    if (S == 0)
    {
        ans -= 1;
    }
    cout << ans << endl;
    return 0;
}
cs


<정답 코드 - 비트마스크>


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
#include<iostream>
#include<vector>
using namespace std;
 
int main()
{
    int N, S, ans = 0;
    vector<int> v;
 
    cin >> N >> S;
 
    v.resize(N);
 
    for (int i = 0; i < N; i++)
    {
        cin >> v[i];
    }
 
    int set = (1 << N) - 1;
 
    for (int subset = set; subset; subset = set & (subset - 1))
    {
       
        int sum = 0;
        for (int i = 0; i < N; i++)
        {
            if ((1 << i)&subset)
            {
                sum += v[i];
            }
        }
 
        if (sum == S)
        {
            ans += 1;
        }
    }
 
    cout << ans << endl;
    
    return 0;
}
cs


반응형

'알고리즘 > BOJ' 카테고리의 다른 글

3015번  (0) 2017.12.19
6549번  (0) 2017.12.18
10974번  (0) 2017.12.14
10973번  (0) 2017.12.14
10972번  (0) 2017.12.14