Codeforces Round #360 (Div. 1) C. The Values You Can Make(DP)

时间:2020-12-21 18:40:47

题目链接:点击打开链接

思路:

用d[i][j][k]表示, 前i个数, 选取的数之和为j, 再次选取的数之和为k的种类数。  那么当这种情况存在时, 我们把此时的k保存即可。

细节参见代码:

#include <cstdio>
#include <cstring>
#include <algorithm>
#include <iostream>
#include <string>
#include <vector>
#include <stack>
#include <bitset>
#include <cstdlib>
#include <cmath>
#include <set>
#include <list>
#include <deque>
#include <map>
#include <queue>
#define Max(a,b) ((a)>(b)?(a):(b))
#define Min(a,b) ((a)<(b)?(a):(b))
using namespace std;
typedef long long ll;
typedef long double ld;
const ld eps = 1e-9, PI = 3.1415926535897932384626433832795;
const int mod = 1000000000 + 7;
const int INF = 0x3f3f3f3f;
// & 0x7FFFFFFF
const int seed = 131;
const ll INF64 = ll(1e18);
const int maxn = 500 + 10;
int T,n,K,cnt[maxn*maxn],d[2][maxn][maxn],kase = 0, a[maxn];
vector<int> res;
int main() {
    scanf("%d%d", &n, &K);
    int sum = 0;
    for(int i = 1; i <= n; i++) {
        scanf("%d", &a[i]);
        sum += a[i];
    }
    int u = 1;
    d[u][0][0] = 1;
    if(sum > K) sum = K;
    for(int i = 1; i <= n; i++) {
        for(int j = 0; j <= sum; j++) {
            for(int k = 0; k <= sum; k++) {
                d[u^1][j][k] += d[u][j][k];
                if(j >= a[i] && k >= a[i]) d[u^1][j][k] += d[u][j-a[i]][k-a[i]];
                if(j >= a[i]) d[u^1][j][k] += d[u][j-a[i]][k];
                if(i == n && d[u^1][j][k]) {
                    if(j == K) {
                        if(!cnt[k]) {
                            res.push_back(k); cnt[k] = 1;
                        }
                    }
                }
            }
        }
        u ^= 1;
    }
    sort(res.begin(), res.end());
    int len = res.size();
    printf("%d\n", len);
    for(int i = 0; i < len; i++) {
        printf("%d%c", res[i], i == len-1 ? '\n' : ' ');
    }
    return 0;
}