hdu母函数

hdu1028

#include <cstdio>
#include <cstring>
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<stdio.h>
#include<string.h>
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<cstdio>
#include<cstring>
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<stdio.h>
#include<string.h>
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<res;i++)
          printf(" %d",vis[i]);
        printf("\n");
    }
    return 0;
}

hdu1521  指数型母函数

#include<stdio.h>
#include<string.h>
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<stdio.h>
#include<string.h>
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;
}

ac代码：

#include<stdio.h>
#include<string.h>
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<cstdio>
#include<cstring>
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;
}