Codeforces Round #360 (Div. 2) E. The Values You Can Make(dp,滚动数组)

题意
给n个数,从中挑若干个元素使其和为k,组成集合S,S的某一子集元素之和为x,问所有x可能的值

思路
dp,状态dp[i][j][p]为考虑到第i个数,当前所有数的和为j,组成和为p的子集是否可能。
如果不使用第i个数,dp[i][j][p]=dp[i-1][j][p]
如果使用第i个数但子集不取他,dp[i][j][p]=dp[i-1][j-ci][p]
如果使用第i个数且子集取他,dp[i][j][p]=dp[i-1][j-ci][p-ci]
但是三维数组是开不下的(偷偷告诉你:其实bool是可以的。。)
所以我们又要用到滚动数组了

#include<bits/stdc++.h>
using namespace  std;
#define ll long long
#define pb push_back
#define inf 2099999999
#define rep(i,a,b) for(int i=a;i<=b;i++)
#define rep1(i,a,b) for(int i=a;i>=b;i--)
const int N=5e2+10;
int arr[N];
int dp[N][N];
/**

*/
int main()
{
    #ifdef LOCAL_DEFINE
        freopen("D://rush.txt", "r", stdin);
    #endif
    ios::sync_with_stdio(false),cin.tie(0);
    int n,k;
    cin>>n>>k;
    rep(i,1,n) cin>>arr[i];
    dp[0][0]=1;
    rep(i,1,n)
    {
        rep1(j,k,arr[i])
        {
            for(int p=j;p>=0;p--)
            {
                if(dp[j-arr[i]][p]||p>=arr[i]&&dp[j-arr[i]][p-arr[i]])
                    dp[j][p]=1;
            }
        }
    }
    int cnt=0;
    for(int i=0;i<=k;i++)
        if(dp[k][i])
            cnt++;
    cout<<cnt<<endl;
    rep(i,0,k)
    if(dp[k][i])
        cout<<i<<' ';
    cout<<endl;
}

posted @ 2018-07-20 18:07  ffgcc  阅读(90)  评论(0编辑  收藏  举报