zoj 3662 Math Magic

题解：

分解质因子后进行状态压缩DP。

由于内存限制，可以使用滚动数组。

一开始用long long竟然给T了。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;
const int mod=1000000007;
int f[1200][100];
int p[1200],pnum,numy;
int a[1200],prime[1200],tprime[1200],label[1200];
void get_prime(int x,int* prime,int kin)
{
    int i=2;if(kin==1)numy=0;
    prime[1]=1;
    if(kin==1){numy++;label[1]=1;}
    while(i*i<=x)
    {
        if(x%i==0)
        {
            if(kin==1)
            {
                numy++;
                label[i]=numy;
            }
            while(x%i==0){x=x/i;prime[i]++;}
        }
        i++;
    }
    if(x>1)
    {
        prime[x]++;
        if(kin==1)
        {
                numy++;
                label[x]=numy;
        }
    }
}
int main()
{
    int n,m,k;
    while(scanf("%d%d%d",&n,&m,&k)!=EOF)
    {
        memset(p,0,sizeof(p));
        memset(f,0,sizeof(f));
        memset(a,0,sizeof(a));
        memset(prime,0,sizeof(prime));
        memset(label,0,sizeof(label));

        get_prime(m,prime,1);
        pnum=0;
        for(int i=1;i<=m;i++)
        if(m%i==0)
        {
            pnum++;
            p[pnum]=i;
            //get(i)
            memset(tprime,0,sizeof(tprime));
            get_prime(i,tprime,0);

            int ans=0;
            for(int j=1;j<=m;j++)
            if(prime[j]>0 && tprime[j]==prime[j])
            ans=ans|(1<<(label[j]-1));

            a[i]=ans;
        }

        f[0][0]=1;
        for(int i=0;i<=k-1;i++)
        {
         for(int j=n;j>=0;j--)
          for(int z=0;z<=(1<<numy)-1;z++)
          {
            for(int s=1;s<=pnum;s++)
            if(j+p[s]<=n)
                f[j+p[s]][z|a[p[s]]]=(f[j+p[s]][z|a[p[s]]]+f[j][z])%mod;
            f[j][z]=0;
          }
        }
        printf("%d\n",f[n][(1<<numy)-1]);
    }
    return 0;
}