zoj 3662 Math Magic

#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
[(1<<numy)-1]);
}
return 0;
}