先脑补一个DP方程:
f[a][b]∗f[c][d]=f[a+c][b∗d Mod m]
那么求
f[n][x]
可以用快速幂求,于是有了一个复杂度为
O(n2logn)
的做法,但是这个复杂度过不了,于是观察一下这个式子,考虑能不能优化。
如果我们能把这个b*d转化成b+d,那么这个式子就可以化成一个卷积的形式,就可以用NTT,因为m是质数,所以其必有原根,我们找到m的原根,把每个数化成原根g的x次方形式,然后就成功把复杂度优化到了
O(nlog2n)
,可以过了
code:
#include<set>
#include<map>
#include<deque>
#include<queue>
#include<stack>
#include<cmath>
#include<ctime>
#include<bitset>
#include<string>
#include<vector>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<complex>
#include<iostream>
#include<algorithm>
#define ll long long
using namespace std;
const int maxn = 20000;
const ll Mod = 1004535809;
int xn,ansn,n,N,ln,id[maxn];
int a[maxn],m,len;
ll g=3,w[maxn];
ll pw(ll x,int k,ll mod)
{
x%=mod;
ll ret=1,tmp=x;
int nowk=0,tmpk=1;
while(nowk!=k)
{
if(tmpk&k)
{
ret=ret*tmp%mod;
nowk|=tmpk;
}
tmpk<<=1;tmp=tmp*tmp%mod;
}
return ret;
}
struct ntt
{
ll s[maxn];
}zero; int _to[maxn];
void NTT(ll *s,int sig)
{
for(int i=0;i<n;i++) if(i<id[i]) swap(s[i],s[id[i]]);
for(int mm=2;mm<=n;mm<<=1)
{
int t=mm>>1, tt=n/mm;
for(int i=0;i<t;i++)
{
ll wn=sig==1?w[i*tt]:w[n-i*tt];
for(int j=i;j<n;j+=mm)
{
ll tx=s[j], ty=s[j+t]*wn%Mod;
s[j]=(tx+ty)%Mod;
s[j+t]=(tx-ty)%Mod;
}
}
}
if(sig==-1) for(int i=0;i<n;i++) s[i]=(s[i]*N%Mod+Mod)%Mod;
}
ntt operator * (ntt x,ntt y)
{
NTT(x.s,1); NTT(y.s,1);
for(int i=0;i<n;i++)x.s[i]=x.s[i]*y.s[i];
NTT(x.s,-1);
for(int i=m;i<n;i++){x.s[_to[i]]+=x.s[i];x.s[i]=0;}
/*ntt ret=zero;
for(int i=0;i<n;i++)
{
for(int j=0;j<=i;j++)
ret.s[i]+=x.s[j]*y.s[i-j];
}
for(int i=m;i<n;i++){ret.s[_to[i]]+=ret.s[i];ret.s[i]=0;}*/
return x;
//return ret;
}
int p[maxn],pn,ind[maxn];
void divide(int x)
{
int t=sqrt(x*1.0);pn=0;
for(int i=2;i<=t;i++)
{
if(x%i==0)
{
p[++pn]=i;
while(x%i==0)x/=i;
}
}
if(x>1)p[++pn]=x;
}
int get_g(int n)
{
divide(n-1);
for(int i=2;i<n;i++)
{
bool flag=true;
for(int j=1;j<=pn;j++)
{
if(pw(i,(n-1)/p[j],n)==1){flag=false; break;}
}
if(flag) return i;
}
}
ntt get_ans(ntt x,int k)
{
ntt ret,tmp=x;
int nowk=0,tmpk=1;
while(nowk!=k)
{
if(tmpk&k)
{
if(nowk==0)ret=tmp;
else ret=ret*tmp;
nowk|=tmpk;
}
tmpk<<=1; tmp=tmp*tmp;
}
return ret;
}
int main()
{
scanf("%d%d%d%d",&ansn,&m,&xn,&len);
for(int i=1;i<=len;i++) scanf("%d",&a[i]);
for(n=1,ln=0;n<(m+m);n<<=1,ln++);
for(int i=0;i<n;i++)id[i]=(id[i>>1]>>1)|((i&1)<<(ln-1));
w[0]=1; w[1]=pw(g,(Mod-1)/n,Mod);
for(int i=2;i<=n;i++)w[i]=w[i-1]*w[1]%Mod;
N=pw(n,Mod-2,Mod);
ll mg=get_g(m);
for(ll tmp=mg,i=1;i<n;i++)
{
if(i<m) ind[tmp]=i;
else _to[i]=ind[tmp];
tmp=tmp*mg%m;
}
ntt st,zero; for(int i=0;i<n;i++) zero.s[i]=0;
st=zero;
for(int i=1;i<=len;i++) if(a[i]!=0) st.s[ind[a[i]]]=1;
ntt ret=get_ans(st,ansn);
printf("%lld\n",(ret.s[ind[xn]]%Mod+Mod)%Mod);
return 0;
}