hdu_1790_The Balance(母函数)

题目连接：http://acm.hdu.edu.cn/showproblem.php?pid=1709

题意：给你一些砝码，让你输出1—sum中不能称出的重量

题解：直接上母函数，在合并括号的时候有加有减，期中mu()为我自己写的模版，所以有点冗杂

#include<cstdio>
#include<cstring>
const int maxn = 10010;
int abs(int a){return a>0?a:-a;}
int c1[maxn], c2[maxn],sum;//c1保存每个指数的系数，c2为中间变量
int data[110],datanum[110],allnum,ans[5000];//硬币类型从0开始,每种类型数量,类型总数
void mu(int n)
{
    for(int i=0;i<=n;i++)c1[i]=0,c2[i]=0;
    for (int i = 0; i*data[0] <= n&&i<=datanum[0]; i++)c1[i*data[0]] = 1;//c1为第一个括号的系数
    for (int i = 1; i <allnum; i++){
        for (int j = 0; j <= n; j++)
            for (int k = 0; k + j <= n&&k<=datanum[i]*data[i]; k += data[i])c2[j + k] += c1[j],c2[abs(j-k)]+=c1[j];//关键在这里
        for (int j = 0; j <= n; j++)c1[j] = c2[j], c2[j] = 0;
    }
}
int main(){
    int t,i;
    for(int ii=0;ii<=105;ii++)datanum[ii]=1;
    while(~scanf("%d",&t)){
        for(i=sum=0,allnum=t;i<t;i++)scanf("%d",&data[i]),sum+=data[i];
        mu(sum);
        int cnt=0;
        for(i=1;i<=sum;i++)if(c1[i]==0)ans[++cnt]=i;
        printf("%d\n",cnt);
        for(int ik=1;ik<=cnt;ik++)printf("%d%c",ans[ik],ik==cnt?'\n':' ');
    }
    return 0;
}