hdu母函数
hdu1028
#include
#include
using namespace std;
const int maxexp=120;
int main()
{
int n, now, pre;
int f[2][maxexp+1];
while (scanf("%d", &n)==1)
{
memset(f, 0, sizeof(f));
for(int i=0; i<=n; i++) //1+x+x^2+...+x^n
f[1%2][i]=1;
for(int i=2; i<=n; i++)
{
now=i%2; pre=(i-1)%2;
for(int j=0; j<=n; j++)
f[now][j]=0;
for(int j=0; j<=n; j++)
for(int k=0; k+j<=n; k+=i) //枚举当前多项式 1+x^i+x^2i+...
f[now][j+k]+=f[pre][j];
}
printf("%d\n", f[n%2][n]);
}
return 0;
} hdu1085
#include
#include
const int maxn=1000+2*1000+5*1000+5;
int c1[maxn],c2[maxn],c3[maxn];
int main()
{
int n1,n2,n3,i,j;
while(scanf("%d%d%d",&n1,&n2,&n3)&&(n1+n2+n3)>0)
{
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
memset(c3,0,sizeof(c3));
for(i=0;i<=n1;i++) c1[i]=1;
for(i=0;i<=n1;i++)
for(j=0;j<=2*n2;j+=2)
c2[i+j]+=c1[i];
for(i=0;i<=n1+2*n2;i++)
for(j=0;j<=5*n3;j+=5)
c3[i+j]+=c2[i];
int flag=0;
for(i=0;i<=n1+2*n2+5*n3;i++)
if(c3[i]==0)
{
printf("%d\n",i);
flag=1; break;
}
if(flag==0) printf("%d\n",n1+2*n2+5*n3+1);
}
return 0;
} hdu2152
#include
#include
using namespace std;
const int maxn=120;
int c1[maxn],c2[maxn];
int num[101][2];
int main()
{
int n,m,i,j,k;
while(~scanf("%d%d",&n,&m))
{
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
for(i=1;i<=n;i++)
scanf("%d%d",&num[i][0],&num[i][1]);
for(i=num[1][0];i<=num[1][1];i++)
c1[i]=1;
for(i=2; i<=n; i++)
{
for(j=num[i][0];j<=num[i][1];j++)
for(k=0;k+j<=m;k++)
c2[j+k]+=c1[k];
for(k=0;k<=m;k++)
c1[k]=c2[k],c2[k]=0;
}
printf("%d\n",c1[m]);
}
return 0;
}
hdu 1709可加，可减。
#include
#include
int num[102];
int sum[10005];
int now[10005];
int vis[10005];
int main()
{
int n,i,j,k;
while(~scanf("%d",&n))
{
int temp=0;
for(i=1;i<=n;i++)
{ scanf("%d",&num[i]);temp+=num[i];}
memset(sum,0,sizeof(sum));
memset(now,0,sizeof(now));
sum[0]=1; sum[num[1]]=1;
for(i=2;i<=n;i++)
{
for(j=0;j<=temp;j++)
for(k=0;k+j<=temp&&k<=num[i];k+=num[i])
{
if(k-j>0) now[k-j]+=sum[j];
else now[j-k]+=sum[j];
now[j+k]+=sum[j];
}
for(k=0;k<=temp;k++)
sum[k]=now[k],now[k]=0;
}
int res=0;
for(i=1;i<=temp;i++)
if(sum[i]==0)
vis[res++]=i;
printf("%d\n",res);
if(res==0) continue;
printf("%d",vis[0]);
for(i=1;i hdu1521 指数型母函数
#include
#include
double jc[]={1,1,2,6,24,120,720,5040,40320,362880,3628800};
double num[12];
double sum[15];
double now[15];
int main()
{
int n,i,j,k,m;
while(~scanf("%d%d",&n,&m))
{
for(i=1;i<=n;i++)
scanf("%lf",&num[i]);
memset(sum,0,sizeof(sum));
memset(now,0,sizeof(now));
for(i=0;i<=num[1];i++)
sum[i]=1.0/jc[i];
for(i=2;i<=n;i++)
{
for(j=0;j<=m;j++)
for(k=0;k+j<=m&&k<=num[i];k++)
now[j+k]+=(sum[j]/jc[k]);
for(k=0;k<=m;k++)
sum[k]=now[k],now[k]=0;
}
double res=sum[m]*1.0*jc[m];
printf("%.0lf\n",res);
}
return 0;
}
poj 3734 Blocks
先用指数型母函数求出前几项，得出结论：res=pow(4,n-1)+pow(2,n-1);
生成函数：
#include
#include
double jc[]={1,1,2,6,24,120,720,5040,40320,362880,3628800};
double num[12];
double sum[15];
double now[15];
int main()
{
int n,i,j,k;
while(~scanf("%d",&n))
{
num[1]=2;num[2]=2;num[3]=1;num[4]=1;
memset(sum,0,sizeof(sum));
memset(now,0,sizeof(now));
for(i=0;i<=n;i+=num[1])
sum[i]=1.0/jc[i];
for(i=2;i<=4;i++)
{
for(j=0;j<=n;j++)
for(k=0;k+j<=n;k+=num[i])
now[j+k]+=(sum[j]/jc[k]);
for(k=0;k<=n;k++)
sum[k]=now[k],now[k]=0;
}
double res=sum[n]*1.0*jc[n];
printf("%.0lf\n",res);
}
return 0;
}
#include
#include
const int mod=10007;
int fun(int n,int k)
{
if(k==1) return n;
if(k==0) return 1;
int res=1;
while(k)
{
if(k%2) res=res*n%mod;
n=(n%mod)*(n%mod)%mod;
k/=2;
}
return res;
}
int main()
{
int ncase,n,res;
scanf("%d",&ncase);
while(ncase--)
{
scanf("%d",&n);
res=fun(4,n-1)+fun(2,n-1);
printf("%d\n",res%mod);
}
return 0;
}
nyoj http://acm.nyist.net/JudgeOnline/problem.php?pid=459
有趣的组合问题
n种颜色，每种颜色有无数种求，拿出m个球有多少种方案数。
母函数：
#include
#include
using namespace std;
#define ll long long
const int maxn=32;
ll c1[maxn],c2[maxn];
int main()
{
//freopen("Input.txt","r",stdin);
//freopen("Out.txt","w",stdout);
int n,m,i,j,k;
while(scanf("%d%d",&n,&m)&&(n||m))
{
memset(c1,0,sizeof(c1));
memset(c2,0,sizeof(c2));
for(i=0; i<=m; i++)
c1[i]=1;
for(i=2; i<=n; i++)
{
for(j=0;j<=m;j++)
for(k=0;k+j<=m;k++)
c2[j+k]+=c1[k];
for(k=0;k<=m;k++)
c1[k]=c2[k],c2[k]=0;
}
printf("%lld\n",c1[m]);
}
return 0;
}