洛咕 P3321 [SDOI2015]序列统计

显然dp就是设(f[i][j])表示dp了i轮，对m取膜是j的方案数

(f[i][xymod m]=f[i-1][x] imes f[i-1][y])

这是(O(nm^2))的

像我这样的蒟蒻都能想到用类似快速幂一样的东西来转移是吧，那么就(O(log_2 nm^2))了

非常难受，还是过不去

如果可以优化一下dp转移就好了，比如把乘改成加，就能用NTT了

然后就要用到一个叫做原根的东西，学NTT的时候只是记了一下不知道这货有啥用

质数(m)原根(g)的性质：对(m-1)进行唯一分解，(m-1=p_1^{a_1}p_2^{a_2}cdots p_k^{a_k})，(g)对任何一个(i)都满足(g^{frac{m-1}{p_i}} eq 1)

然后，这个题要用到的就是，(g^0,g^1,cdots,g^{m-2})互不相同

那么，把原来f[1][i]变成f[1][g^{i-1}]，就可以一一对应了，又把乘变成了加

#include<bits/stdc++.h>
#define il inline
#define vd void
#define int ll
#define mod 1004535809
typedef long long ll;
il int gi(){
    int x=0,f=1;
    char ch=getchar();
    while(!isdigit(ch)){
        if(ch=='-')f=-1;
        ch=getchar();
    }
    while(isdigit(ch))x=x*10+ch-'0',ch=getchar();
    return x*f;
}
il int Pow(int x,int y,int mo){
    int ret=1;
    while(y){
        if(y&1)ret=ret*x%mo;
        x=x*x%mo;y>>=1;
    }
    return ret;
}
const int G=3,iG=Pow(G,mod-2,mod);
int n,m,qx,k,F[8000];
struct yyb{
    ll f[8000];
    yyb(){memset(f,0,sizeof f);}
};
ll N,lg,rev[1<<14],P[1<<14],iP[1<<14],invN;
il vd ntt(ll*A,int n,int t){
    for(int i=0;i<n;++i)if(rev[i]>i)std::swap(A[i],A[rev[i]]);
    for(int o=1;o<n;o<<=1){
        int W=t?P[o]:iP[o];
        for(int*p=A;p!=A+n;p+=o<<1)
            for(int i=0,w=1;i<o;++i,w=w*W%mod){
                int t=w*p[i+o]%mod;
                p[i+o]=(p[i]-t+mod)%mod;p[i]=(p[i]+t)%mod;
            }
    }
    if(!t)for(int i=0;i<n;++i)A[i]=A[i]*invN%mod;
}
il yyb mul(const yyb&a,const yyb&b){
    static int A[1<<14],B[1<<14];
    memset(A,0,N*8),memset(B,0,N*8);
    memcpy(A,a.f,sizeof a.f),memcpy(B,b.f,sizeof b.f);
    ntt(A,N,1),ntt(B,N,1);
    for(int i=0;i<N;++i)A[i]=A[i]*B[i]%mod;
    ntt(A,N,0);
    yyb ret;
    for(int i=0;i<=m*2;++i)(ret.f[i%(m-1)]+=A[i])%=mod;
    return ret;
}
signed main(){
    n=gi(),m=gi(),qx=gi(),k=gi();	
    int gM;
    {//get gM
        int p[10],tot=0;
        for(int x=m-1,i=2;i<=x;++i)
            if(x%i==0){
                p[++tot]=i;
                while(x%i==0)x/=i;
            }
        for(int i=2;i<=m;++i){
            for(int j=1;j<=tot;++j)if(Pow(i,(m-1)/p[j],m)==1)goto gg;
            gM=i;break;gg:;
        }
        for(int i=0,j=1;i<m-1;++i,j=j*gM%m)F[j]=i;
    }
    N=1,lg=0;while(N<m<<1)N<<=1,++lg;
    for(int i=0;i<N;++i)rev[i]=(rev[i>>1]>>1)|((i&1)<<lg-1);
    for(int i=1;i<N;i<<=1)P[i]=Pow(G,(mod-1)/(i<<1),mod),iP[i]=Pow(iG,(mod-1)/(i<<1),mod);
    invN=Pow(N,mod-2,mod);
    yyb X,ans;
    int t;
    while(k--){
        t=gi();
        if(t)X.f[F[t]]=1;
    }
    ans.f[F[1]]=1;
    while(n){
        if(n&1)ans=mul(ans,X);
        X=mul(X,X);n>>=1;
    }
    printf("%lld
",ans.f[F[qx]]);
    return 0;
}