BZOJ3992: [SDOI2015]序列统计

先脑补一个DP方程：$f[a][b]*f[c][d]=f[a+c][b*d\ Mod\ m]$
那么求$f[n][x]$可以用快速幂求，于是有了一个复杂度为$O(n^2logn)$的做法，但是这个复杂度过不了，于是观察一下这个式子，考虑能不能优化。
如果我们能把这个b*d转化成b+d，那么这个式子就可以化成一个卷积的形式，就可以用NTT，因为m是质数，所以其必有原根，我们找到m的原根，把每个数化成原根g的x次方形式，然后就成功把复杂度优化到了$O(nlog^2n)$，可以过了

---

---

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;
}