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;
}