티스토리 뷰

www.acmicpc.net/problem/1208

 

1208번: 부분수열의 합 2

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

www.acmicpc.net

백준 소스코드 [C++] 1208 부분수열의 합 2

#include <iostream>
#include <algorithm>
#include <queue>
#include <string.h>
#include <limits.h>
#include <vector>
#include <math.h>
#include <stack>
#include <bitset>
#include <string>
#include <set>
#include <map>
#include <unordered_map>
#include <sstream>
#include <cstdlib>
#define all(v) v.begin(), v.end()
#define pii pair<int, int>
#define pli pair<ll, int>
#define make_unique(v) sort(all(v)), v.erase(unique(all(v)), v.end())
typedef long long ll;
using namespace std;

int n,k;
vector<int> v;
map <int, int>leftMap;
ll res = 0LL;

void memo1(int start, int sum) {
    if (start == n/2) {
        leftMap[sum]++;
        return;
    }
    memo1(start + 1, sum);
    memo1(start + 1, sum + v[start]);
}

void memo2(int mid, int sum) {
    if (mid == n) {
        res += leftMap[sum];
        return;
    }
    memo2(mid + 1, sum);
    memo2(mid + 1, sum - v[mid]);
}

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

    cin >> n >> k;
    v.resize(n);
    for (int i = 0;i < n;i++) cin >> v[i];

    memo1(0, 0);
    memo2(n / 2, k);
    if (k == 0) res--;
    cout << res;

    return 0;
}

 

 

댓글