usaco 2.1.4 Subset Sums 【母函数】

(1+x)*(1+x^2)*(1+x^3)*……*x^n   则 x^(n*(n+1)/2) 的系数一半为ans

/*
 ID:   w.x.f.g1
 PROG: subset
 LANG: C++
*/

#include<iostream>
#include<cstring>
#include<fstream>
using namespace std;
typedef long long ll;
ll c1[400],c2[400],c3[400];
int main()
{
    ifstream cin("subset.in");
    ofstream cout("subset.out");
    int n;cin>>n;
    int sum=n*(n+1)/2;
    if(sum&1){cout<<0<<endl;return 0;}
    c1[0]=c1[1]=1;
    for(int i=2;i<=n;i++)
    {
        c3[0]=0,c3[1]=i;
        memset(c2,0,sizeof(c2));
        for(int j=0;j<=sum>>1;j++)
            for(int k=0;k<2&&c3[k]+1<=sum>>1;k++)
                c2[c3[k]+j]+=c1[j];
        copy(c2,c2+sum/2+1,c1);
    }
    cout<<c1[sum>>1]/2<<endl;
}

你可能感兴趣的:(c)